Searched refs:DeviceTypeString (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | types_test.cc | 26 EXPECT_EQ("CPU", DeviceTypeString(DeviceType(DEVICE_CPU))); in TEST()
27 EXPECT_EQ("GPU", DeviceTypeString(DeviceType(DEVICE_GPU))); in TEST()
28 EXPECT_EQ("SYCL", DeviceTypeString(DeviceType(DEVICE_SYCL))); in TEST()
|
| D | types.cc | 201 string DeviceTypeString(const DeviceType& device_type) { in DeviceTypeString() function
|
| D | op_kernel.cc | 903 return strings::StrCat(op_type, ":", DeviceTypeString(device_type), ":", in Key()
1030 DeviceTypeString(device_type), " devices compatible with node ", in FindKernelDef()
1134 "' OpKernel for ", DeviceTypeString(device_type), in CreateOpKernel()
|
| D | types.h | 109 string DeviceTypeString(const DeviceType& device_type);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 61 DeviceTypeString(context->device_type()))); in AvgPoolingOp()
220 DeviceTypeString(context->device_type()))); in AvgPoolingGradOp()
|
| D | pooling_ops_3d.cc | 143 DeviceTypeString(context->device_type()))); in Pooling3DOp()
316 "on device type ", DeviceTypeString(context->device_type()))); in MaxPooling3dGradOp()
470 "on device type ", DeviceTypeString(context->device_type()))); in AvgPooling3dGradOp()
|
| D | maxpooling_op.cc | 209 DeviceTypeString(context->device_type()))); in MaxPoolingGradOp()
448 "on device type ", DeviceTypeString(context->device_type()))); in MaxPoolingGradGradOp()
751 DeviceTypeString(context->device_type()))); in MaxPoolingNoMaskOp()
804 DeviceTypeString(context->device_type()))); in MaxPoolingNoMaskV2Op()
|
| D | pooling_ops_common.h | 91 DeviceTypeString(context->device_type()))); in MaxPoolingOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.cc | 481 strings::StrAppend(&devices_registered, DeviceTypeString(device_type), in DebugInfo()
|