| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | set_device.cc | 23 Status SetDevice(const GraphDef& input_graph_def, in SetDevice() function
44 REGISTER_GRAPH_TRANSFORM("set_device", SetDevice);
|
| D | set_device_test.cc | 32 Status SetDevice(const GraphDef& input_graph_def,
90 TF_ASSERT_OK(SetDevice(graph_def, context, &result)); in TEST()
113 TF_ASSERT_OK(SetDevice(graph_def, context, &result)); in TEST()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | grpc_session_debug_test.cc | 215 void SetDevice(GraphDef* graph, const string& name, const string& dev) { in SetDevice() function
247 SetDevice(&def, a->name(), a_dev.name()); in TEST_F()
248 SetDevice(&def, b->name(), b_dev.name()); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | cluster.h | 60 void SetDevice(const string& device) { device_ = device; } in SetDevice() function
|
| D | resolve_svdf.cc | 273 cluster->SetDevice(node.device()); in CreateCluster()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 399 void SetDevice(GraphDef* graph, const string& name, const string& dev) { in SetDevice() function
440 SetDevice(&def, a->name(), a_dev.name()); in TEST()
441 SetDevice(&def, b->name(), b_dev.name()); in TEST()
442 SetDevice(&def, c->name(), c_dev.name()); in TEST()
512 SetDevice(&def, fill_node->name(), cluster->devices()[0].name()); in TEST()
513 SetDevice(&def, fill_node->name(), cluster->devices()[1].name()); in TEST()
552 SetDevice(&def, a->name(), a_dev.name()); in TEST()
553 SetDevice(&def, b->name(), b_dev.name()); in TEST()
605 SetDevice(&def, a->name(), src.name()); in TEST()
606 SetDevice(&def, b->name(), dst0.name()); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_ops_test.cc | 43 SetDevice(DEVICE_GPU, in Run()
|
| D | ops_testutil.h | 76 void SetDevice(const DeviceType& device_type, std::unique_ptr<Device> device);
|
| D | ops_testutil.cc | 99 void OpsTestBase::SetDevice(const DeviceType& device_type, in SetDevice() function in tensorflow::OpsTestBase
|
| D | non_max_suppression_op_gpu_test.cc | 45 SetDevice(DEVICE_GPU, in MakeOp()
|
| D | resize_bilinear_op_test.cc | 47 SetDevice(DEVICE_GPU, std::move(device_gpu)); in SetUp()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_composition_test.cc | 53 SetDevice(DeviceType(xla_device_name), in RunComposedOp()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 124 op.SetDevice(send_device_); in StartSend()
258 op.SetDevice(recv_device_); in StartRecv()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 468 op->SetDevice(device); in EagerLocalExecute()
637 op->SetDevice(device); in EagerRemoteExecute()
849 op->SetDevice(resource_device); in MaybeUpdateOpDevice()
892 op->SetDevice(ctx.HostCPU()); in MaybeUpdateOpDevice()
|
| D | eager_operation.h | 79 void SetDevice(tensorflow::Device* device) { in SetDevice() function
|
| D | mkl_eager_op_rewrite.cc | 139 (*new_mkl_op)->SetDevice(orig_op->Device()); in SetupNewOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 75 OpsTestBase::SetDevice(DEVICE_GPU, std::move(device)); in AddSimpleTrtOp()
|
| D | trt_engine_resource_ops_test.cc | 97 SetDevice(DEVICE_GPU, std::move(device)); in TEST_F()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 350 OpsTestBase::SetDevice(DEVICE_GPU, std::move(device)); in SetupOp()
|
| /external/skqp/src/compute/hs/cuda/bench/ |
| D | main.c | 499 cuda(SetDevice(device)); in main()
|
| /external/stressapptest/src/ |
| D | worker.h | 681 virtual void SetDevice(const char *device_name);
|
| D | sat.cc | 1326 thread->SetDevice(diskfilename_[i].c_str()); in InitializeThreads()
1343 rthread->SetDevice(diskfilename_[i].c_str()); in InitializeThreads()
|
| D | worker.cc | 2605 void DiskThread::SetDevice(const char *device_name) { in SetDevice() function in DiskThread
|
| /external/toolchain-utils/android_bench_suite/ |
| D | autotest.diff | 561 +NAME = "SetDevice"
907 + set_device = ['SetDevice']
|