HomeSort by relevance Sort by last modified time
    Searched refs:assigned_device_name_index (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_rewrite.cc 53 int recv_dev = node->assigned_device_name_index();
63 int send_dev = src_node->assigned_device_name_index();
118 int send_dev = node->assigned_device_name_index();
127 : edge->dst()->assigned_device_name_index();
  /external/tensorflow/tensorflow/core/common_runtime/
placer.cc 794 assigned_device = input->assigned_device_name_index();
    [all...]
  /external/tensorflow/tensorflow/core/graph/
graph.h 118 int assigned_device_name_index() const { return assigned_device_name_index_; } function in class:tensorflow::Node
553 return device_names_[node.assigned_device_name_index()];

Completed in 126 milliseconds