| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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 | 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 | 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 | 264 std::unique_ptr<KernelDef> kdef(new KernelDef); in RegisterCompilationKernels() local
265 kdef->set_op(op_registration->name); in RegisterCompilationKernels()
266 kdef->set_device_type(backend.first); in RegisterCompilationKernels()
267 kdef->set_label(op_registration->label); in RegisterCompilationKernels()
275 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); in RegisterCompilationKernels()
325 !backend.second.op_filter(kdef.get())) { in RegisterCompilationKernels()
332 new KernelDef(*kdef), "XlaJitOp", op_registration->factory)); in RegisterCompilationKernels()
333 backend.second.kernel_defs.push_back(std::move(kdef)); in RegisterCompilationKernels()
|
| D | tf2xla_util.h | 60 KernelDef* kdef);
|
| D | xla_op_registry.h | 139 typedef bool (*BackendOpFilter)(KernelDef* kdef);
|
| D | tf2xla_util.cc | 528 KernelDef* kdef) { in AddDtypeToKernelDefConstraint() argument
529 for (KernelDef::AttrConstraint& constraint : *kdef->mutable_constraint()) { in AddDtypeToKernelDefConstraint()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_node_device_util.cc | 24 bool TpuOpFilter(KernelDef* kdef) { in TpuOpFilter() argument
25 StringPiece op(kdef->op()); in TpuOpFilter()
29 AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef); in TpuOpFilter()
32 AddDtypeToKernelDefConstraint("T", DT_STRING, kdef); in TpuOpFilter()
|
| D | tpu_node_device_util.h | 26 bool TpuOpFilter(KernelDef* kdef);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.cc | 91 const KernelDef* kdef = nullptr; in MemoryTypesForNode() local
93 FindKernelDef(device_type, ndef, &kdef, nullptr /* kernel_class_name */); in MemoryTypesForNode()
127 const auto& from_proto = kdef->host_memory_arg(); in MemoryTypesForNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 74 const NodeDef& node, const KernelDef** kdef) { in TryFindKernelDef() argument
79 if (kdef) { in TryFindKernelDef()
80 *kdef = kernel; in TryFindKernelDef()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 307 const KernelDef* kdef; in CreateCopyNode() local
343 Status s = FindKernelDef(device_type, node_def, &kdef, nullptr); in CreateCopyNode()
477 const KernelDef* kdef; in CreateDebugNode() local
497 if (!FindKernelDef(DeviceType(device.device_type()), node_def, &kdef, nullptr) in CreateDebugNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 560 const KernelDef* kdef; in ReplaceTensorWithConstant() local
561 if (!FindKernelDef(device_type, def, &kdef, nullptr).ok()) { in ReplaceTensorWithConstant()
|