| /external/tensorflow/tensorflow/python/util/ |
| kernel_registry.cc | 52 tensorflow::FindKernelDef(tensorflow::DeviceType(parsed_name.type.c_str()),
|
| /external/tensorflow/tensorflow/core/framework/ |
| memory_types.cc | 81 FindKernelDef(device_type, ndef, &kdef, nullptr /* kernel_class_name */);
|
| op_kernel.h | [all...] |
| op_kernel.cc | [all...] |
| op_kernel_test.cc | 474 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name);
[all...] |
| /external/tensorflow/tensorflow/core/debug/ |
| debug_graph_utils.cc | 310 Status s = FindKernelDef(device_type, node_def, &kdef, nullptr);
465 if (!FindKernelDef(DeviceType(device.device_type()), node_def, &kdef, nullptr)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| constant_folding.cc | 248 if (!FindKernelDef(DeviceType(DEVICE_CPU), n->def(), /*def=*/nullptr,
524 if (!FindKernelDef(device_type, def, &kdef, nullptr).ok()) {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_op_registry.cc | 47 TF_RETURN_IF_ERROR(FindKernelDef(device_type, node_def, /*KernelDef*/ nullptr,
|
| /external/tensorflow/tensorflow/python/grappler/ |
| cluster.i | 243 tensorflow::Status s = tensorflow::FindKernelDef(dev_type, node, nullptr, nullptr);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| mark_for_compilation_pass.cc | 53 return FindKernelDef(jit_device_type, node.def(), nullptr, nullptr).ok();
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | 804 status = FindKernelDef(test_device_type, *node_def, nullptr, nullptr);
[all...] |
