Searched refs:kdef (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_gpu_backend.cc | 22 bool GpuOpFilter(KernelDef* kdef) { in GpuOpFilter() argument
23 if (kdef->op() == "Const") { in GpuOpFilter()
24 AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef); in GpuOpFilter()
26 if (kdef->op() == "Assert") { in GpuOpFilter()
27 AddDtypeToKernelDefConstraint("T", DT_STRING, kdef); in GpuOpFilter()
|
| D | xla_cpu_backend.cc | 22 bool CpuOpFilter(KernelDef* kdef) { in CpuOpFilter() argument
23 if (kdef->op() == "Const") { in CpuOpFilter()
24 AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef); in CpuOpFilter()
26 if (kdef->op() == "Assert") { in CpuOpFilter()
27 AddDtypeToKernelDefConstraint("T", DT_STRING, kdef); in CpuOpFilter()
|
| D | tf2xla_supported_ops.cc | 48 for (const KernelDef* kdef : kdefs) { in PrintSupportedOps() local
50 for (const KernelDef::AttrConstraint& constraint : kdef->constraint()) { in PrintSupportedOps()
59 std::cout << "`" << kdef->op() << "` | " in PrintSupportedOps()
|
| D | xla_op_registry.cc | 253 std::unique_ptr<KernelDef> kdef(new KernelDef); in RegisterCompilationKernels() local
254 kdef->set_op(op_registration->name); in RegisterCompilationKernels()
255 kdef->set_device_type(backend.first); in RegisterCompilationKernels()
263 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); in RegisterCompilationKernels()
310 !backend.second.op_filter(kdef.get())) { in RegisterCompilationKernels()
317 new KernelDef(*kdef), "XlaJitOp", op_registration->factory)); in RegisterCompilationKernels()
318 backend.second.kernel_defs.push_back(std::move(kdef)); in RegisterCompilationKernels()
|
| D | tf2xla_util.h | 59 KernelDef* kdef);
|
| D | xla_op_registry.h | 104 typedef bool (*BackendOpFilter)(KernelDef* kdef);
|
| D | tf2xla_util.cc | 529 KernelDef* kdef) { in AddDtypeToKernelDefConstraint() argument
530 for (KernelDef::AttrConstraint& constraint : *kdef->mutable_constraint()) { in AddDtypeToKernelDefConstraint()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.cc | 84 const KernelDef* kdef = nullptr; in MemoryTypesForNode() local
86 FindKernelDef(device_type, ndef, &kdef, nullptr /* kernel_class_name */); in MemoryTypesForNode()
118 const auto& from_proto = kdef->host_memory_arg(); in MemoryTypesForNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 72 const NodeDef& node, const KernelDef** kdef) { in TryFindKernelDef() argument
77 if (kdef) { in TryFindKernelDef()
78 *kdef = kernel; in TryFindKernelDef()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 274 const KernelDef* kdef; in CreateCopyNode() local
310 Status s = FindKernelDef(device_type, node_def, &kdef, nullptr); in CreateCopyNode()
444 const KernelDef* kdef; in CreateDebugNode() local
464 if (!FindKernelDef(DeviceType(device.device_type()), node_def, &kdef, nullptr) in CreateDebugNode()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_interpreter_device.cc | 78 static bool OpFilter(KernelDef* kdef) { return true; } in OpFilter() argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 533 const KernelDef* kdef; in ReplaceTensorWithConstant() local
534 if (!FindKernelDef(device_type, def, &kdef, nullptr).ok()) { in ReplaceTensorWithConstant()
|