| /aosp_15_r20/external/tensorflow/tensorflow/python/util/ |
| H A D | kernel_registry.cc | 52 status = tensorflow::FindKernelDef( in TryFindKernelClass()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | debug_graph_utils.cc | 343 Status s = FindKernelDef(device_type, node_def, &kdef, nullptr); in CreateCopyNode()
497 if (!FindKernelDef(DeviceType(device.device_type()), node_def, &kdef, nullptr) in CreateDebugNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/ |
| H A D | list_flex_ops.cc | 69 if (!tensorflow::FindKernelDef( in FindTensorflowKernelClass()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel_test_base.h | 156 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name);
|
| H A D | memory_types.cc | 93 FindKernelDef(device_type, ndef, &kdef, nullptr /* kernel_class_name */); in MemoryTypesForNode()
|
| H A D | op_kernel.cc | 1471 Status FindKernelDef( in FindKernelDef() function
1513 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef() function
1515 return FindKernelDef( in FindKernelDef()
|
| H A D | op_kernel.h | 1467 Status FindKernelDef(
1477 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | pin_to_host_optimizer.cc | 76 Status s = FindKernelDef(device, node, &kernel, nullptr); in TryFindKernelDef()
|
| H A D | generic_layout_optimizer_transposer.cc | 143 Status s = FindKernelDef(device_type, node, nullptr, nullptr); in IsHostMemory()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | partially_decluster_pass.cc | 258 *must_compile = !FindKernelDef(device_type, n->def(), nullptr, nullptr).ok(); in MustCompileNode()
|
| H A D | compilability_check_util.cc | 203 Status s = FindKernelDef(jit_device_type_, node.def(), nullptr, nullptr); in HasXLAKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
| H A D | cluster_wrapper.cc | 189 tensorflow::FindKernelDef(dev_type, node, nullptr, nullptr); in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | eval_const_tensor.cc | 275 return FindKernelDef(DeviceType(DEVICE_CPU), node.def(), /*def=*/nullptr, in HasCpuKernel()
|
| H A D | constant_folding.cc | 558 if (!FindKernelDef(device_type, def, &kdef, nullptr).ok()) { in ReplaceTensorWithConstant()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | utils.cc | 487 return FindKernelDef(DeviceType(parsed_name.type), node_name, in IsKernelRegisteredForNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_op_registry.cc | 49 TF_RETURN_IF_ERROR(FindKernelDef(device_type, node_def, /*KernelDef*/ nullptr, in LaunchOpHasKernelForDevice()
|
| H A D | xla_compiler.cc | 1294 s = FindKernelDef(device_type, node->def(), nullptr, nullptr); in ValidateGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_experimental.cc | 447 status->status = tensorflow::FindKernelDef( in TF_AttrBuilderCheckCanRunOnDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | execute.cc | 300 Status s = FindKernelDef(DeviceType(op_device->device_type()), *node_def, in GetKernelDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 1514 status = FindKernelDef(test_device_type, *node_def, nullptr, nullptr); in ExpectTfAndXlaOutputsAreClose()
|