Home
last modified time | relevance | path

Searched refs:DummyDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc25 class DummyDevice : public DeviceBase { class
27 DummyDevice(Env* env, bool save) : DeviceBase(env), save_(save) {} in DummyDevice() function in tensorflow::__anon7be2f0650111::DummyDevice
61 DummyDevice dummy_device(nullptr, false); in TestBitcastOp()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding_test.cc33 class DummyDevice : public DeviceBase { class
35 explicit DummyDevice(Env* env) : DeviceBase(env) { in DummyDevice() function in tensorflow::DummyDevice
92 DummyDevice cpu_device(Env::Default()); in Validate()
179 DummyDevice cpu_device(Env::Default()); in BM_TensorResponse()
/external/tensorflow/tensorflow/c/
Dkernels_test.cc128 class DummyDevice : public DeviceBase { class
130 DummyDevice(Env* env, bool save) : DeviceBase(env), save_(save) {} in DummyDevice() function in tensorflow::DummyDevice
198 DummyDevice dummy_device(nullptr, false); in TEST()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc309 class DummyDevice : public DeviceBase { class
311 DummyDevice(Env* env, bool save) : DeviceBase(env), save_(save) {} in DummyDevice() function in tensorflow::__anon843475bc0111::DummyDevice
325 params.device = new DummyDevice(env, params.record_tensor_accesses); in TEST_F()
350 params.device = new DummyDevice(env, params.record_tensor_accesses); in TEST_F()
378 params.device = new DummyDevice(env, params.record_tensor_accesses); in TEST_F()
659 std::unique_ptr<DummyDevice> device( in TEST_F()
660 new DummyDevice(env, params.record_tensor_accesses)); in TEST_F()
936 std::unique_ptr<DummyDevice> device(new DummyDevice(Env::Default(), false)); in BM_InputRangeHelper()