Searched refs:FindKernelDef (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/python/util/ |
| D | kernel_registry.cc | 52 tensorflow::FindKernelDef(tensorflow::DeviceType(parsed_name.type.c_str()), in TryFindKernelClass()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.cc | 86 FindKernelDef(device_type, ndef, &kdef, nullptr /* kernel_class_name */); in MemoryTypesForNode()
|
| D | op_kernel_test.cc | 507 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name);
|
| D | op_kernel.h | 1421 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def,
|
| D | op_kernel.cc | 1147 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef() function
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 310 Status s = FindKernelDef(device_type, node_def, &kdef, nullptr); in CreateCopyNode()
464 if (!FindKernelDef(DeviceType(device.device_type()), node_def, &kdef, nullptr) in CreateDebugNode()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 248 *must_compile = !FindKernelDef(device_type, n->def(), nullptr, nullptr).ok(); in MustCompileNode()
|
| D | mark_for_compilation_pass.cc | 140 return FindKernelDef(jit_device_type, node.def(), nullptr, nullptr).ok(); in HasXLAKernel()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 75 Status s = FindKernelDef(device, node, &kernel, nullptr); in TryFindKernelDef()
|
| D | layout_optimizer.cc | 377 Status s = FindKernelDef(device_type, node, nullptr, nullptr); in IsHostMemory()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 118 return FindKernelDef(DeviceType(DEVICE_CPU), node.def(), /*def=*/nullptr, in HasCpuKernel()
|
| D | constant_folding.cc | 534 if (!FindKernelDef(device_type, def, &kdef, nullptr).ok()) { in ReplaceTensorWithConstant()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 498 return FindKernelDef(DeviceType(parsed_name.type), node, nullptr, nullptr); in IsKernelRegisteredForNode()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 250 tensorflow::Status s = tensorflow::FindKernelDef(dev_type, node, nullptr, nullptr); in TF_GetSupportedDevices()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.cc | 49 TF_RETURN_IF_ERROR(FindKernelDef(device_type, node_def, /*KernelDef*/ nullptr, in LaunchOpHasKernelForDevice()
|
| D | xla_compiler.cc | 1011 s = FindKernelDef(device_type, node->def(), nullptr, nullptr); in ValidateGraph()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 8856 status->status = tensorflow::FindKernelDef( in TF_AttrBuilderCheckCanRunOnDevice()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 892 status = FindKernelDef(test_device_type, *node_def, nullptr, nullptr); in ExpectTfAndXlaOutputsAreClose()
|