| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| sharding_util.cc | 93 device_name = node.requested_device();
103 device_name = src.requested_device();
|
| tf2xla_util.cc | 279 n->set_requested_device(matching_node->requested_device());
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| placer.cc | 344 if (DeviceNameUtils::ParseFullName(node->requested_device(),
364 node->requested_device(), " but available devices are [ ",
370 node->requested_device(), "' because no supported kernel for ",
377 node->requested_device(), debug_info);
384 node->requested_device(),
560 if (!node.requested_device().empty()) {
566 if (!DeviceNameUtils::ParseFullName(node.requested_device(),
569 node.requested_device(), "'");
[all...] |
| parallel_concat_optimizer.cc | 56 node_builder.Device(n->requested_device());
|
| accumulate_n_optimizer.cc | 82 node_builder.Device(n->requested_device());
|
| graph_execution_state.cc | 368 node->set_assigned_device_name(node->requested_device());
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| build_xla_launch_ops_pass.cc | 86 node->requested_device(), const_dtypes, num_resource_args, arg_dtypes,
|
| encapsulate_subgraphs_pass.cc | 643 ? node->requested_device()
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| session_ops.cc | 46 TensorStore::TensorAndKey tk{val, id, requested_device()};
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition_test.cc | 84 string device_name = !node->requested_device().empty()
85 ? node->requested_device()
|
| graph.h | 101 const string& requested_device() const;
|
| graph.cc | 161 const string& Node::requested_device() const { return def().device(); } function in class:tensorflow::Node
|
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.h | 124 const string& requested_device() const; // Same as def().device()
[all...] |
| op_kernel.cc | 118 const string& OpKernel::requested_device() const { return def_->device(); } function in class:tensorflow::OpKernel
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | [all...] |
