Searched refs:DeviceTypeString (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | types_test.cc | 27 EXPECT_EQ("CPU", DeviceTypeString(DeviceType(DEVICE_CPU))); in TEST()
28 EXPECT_EQ("GPU", DeviceTypeString(DeviceType(DEVICE_GPU))); in TEST()
29 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 | 1062 return strings::StrCat(op_type, ":", DeviceTypeString(device_type), ":", in Key()
1156 DeviceTypeString(device_type), " devices compatible with node ", in FindKernelDef()
1291 "' OpKernel for ", DeviceTypeString(device_type), in CreateOpKernel()
|
| D | types.h | 112 string DeviceTypeString(const DeviceType& device_type);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 61 DeviceTypeString(context->device_type()))); in AvgPoolingOp()
227 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 | 223 DeviceTypeString(context->device_type()))); in MaxPoolingGradOp()
462 "on device type ", DeviceTypeString(context->device_type()))); in MaxPoolingGradGradOp()
765 DeviceTypeString(context->device_type()))); in MaxPoolingNoMaskOp()
818 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 | colocation_graph.cc | 117 v.push_back(DeviceTypeString(device_and_type.first)); in DeviceTypeAndPriorityToString()
885 DeviceTypeString(device_type.first), " "); in DebugInfo()
|