| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | implementation_selector_test.cc | 39 constexpr char CpuDevice[] = "/device:CPU:0"; variable
46 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {CpuDevice}); in TEST_F()
72 CpuDevice), in TEST_F()
96 CpuDevice), in TEST_F()
121 CpuDevice), in TEST_F()
152 CpuDevice), in TEST_F()
176 CpuDevice), in TEST_F()
213 CpuDevice), in TEST_F()
220 NDef("y2", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, CpuDevice), in TEST_F()
221 NDef("z2", "Identity", {"y2"}, {{"T", DT_FLOAT}}, CpuDevice)}, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op_cpu_impl.h | 27 using CpuDevice = Eigen::ThreadPoolDevice; variable
30 template struct functor::MirrorPad<CpuDevice, T, int32, CPU_PROVIDED_IXDIM>; \
31 template struct functor::MirrorPad<CpuDevice, T, int64, CPU_PROVIDED_IXDIM>;
37 template struct functor::MirrorPadGrad<CpuDevice, T, int32, \
39 template struct functor::MirrorPadGrad<CpuDevice, T, int64, \
|
| D | mirror_pad_op.cc | 149 using CpuDevice = Eigen::ThreadPoolDevice; typedef
157 void MirrorPad<CpuDevice, T, Tpaddings, i>::operator()( \
158 const CpuDevice&, typename TTypes<T, i, int32>::Tensor, \
161 extern template struct MirrorPad<CpuDevice, T, Tpaddings, i>;
188 MirrorPadOp<CpuDevice, type, int32>); \
194 MirrorPadOp<CpuDevice, type, int64>);
366 void MirrorPadGrad<CpuDevice, T, Tpaddings, k>::operator()( \
367 const CpuDevice&, typename TTypes<T, k, int32>::Tensor, \
371 extern template struct MirrorPadGrad<CpuDevice, T, Tpaddings, k>;
396 MirrorPadGradOp<CpuDevice, type, int32>); \
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | cpu_device.cc | 27 CpuDevice::CpuDevice(int id, in CpuDevice() function in xla::CpuDevice
56 auto device = absl::make_unique<CpuDevice>(i, std::move(device_state)); in GetCpuClient()
|
| D | cpu_device.h | 26 class CpuDevice : public PjRtStreamExecutorDevice {
28 CpuDevice(int id, std::unique_ptr<LocalDeviceState> local_device_state);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op_cpu_impl.h | 27 using CpuDevice = Eigen::ThreadPoolDevice; variable
30 template struct functor::Slice<CpuDevice, T, CPU_PROVIDED_IXDIM>;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base_test.cc | 28 TEST(DeviceBaseTest, CpuDevice) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 154 py::class_<CpuDevice, PjRtDevice, ClientAndPtr<CpuDevice>>(m, "CpuDevice") in PYBIND11_MODULE()
155 .def("__repr__", [](const CpuDevice& device) { in PYBIND11_MODULE()
|
| D | outfeed_receiver_test.cc | 97 devices.push_back(absl::make_unique<CpuDevice>(0, std::move(device_state))); in GetCpuClientWithNonLocalDevice()
98 devices.push_back(absl::make_unique<CpuDevice>(1, nullptr)); in GetCpuClientWithNonLocalDevice()
|