| /external/tensorflow/tensorflow/core/framework/ |
| D | kernel_def_builder_test.cc | 27 const KernelDef* def = KernelDefBuilder("A").Device(DEVICE_CPU).Build(); in TEST()
28 KernelDef expected; in TEST()
36 const KernelDef* def = KernelDefBuilder("B") in TEST()
40 KernelDef expected; in TEST()
77 const KernelDef* def = KernelDefBuilder("B") in TEST()
81 KernelDef expected; in TEST()
118 const KernelDef* def = KernelDefBuilder("B") in TEST()
122 KernelDef expected; in TEST()
159 const KernelDef* def = KernelDefBuilder("E") in TEST()
164 KernelDef expected; in TEST()
|
| D | kernel_def_builder.h | 27 class KernelDef; variable
84 const KernelDef* Build();
87 KernelDef* kernel_def_;
|
| D | kernel_def_builder.cc | 23 kernel_def_ = new KernelDef; in KernelDefBuilder()
144 const KernelDef* KernelDefBuilder::Build() { in Build()
145 KernelDef* r = kernel_def_; in Build()
|
| D | kernel_def_util_test.cc | 33 KernelDef KernelDefFromText(const string& text) { in KernelDefFromText()
34 KernelDef kernel_def; in KernelDefFromText()
|
| D | kernel_def.proto | 13 message KernelDef { message
47 repeated KernelDef kernel = 1;
|
| D | kernel_def_util.h | 26 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs,
|
| D | op_kernel.cc | 1120 KernelRegistration(const KernelDef& d, StringPiece c, in KernelRegistration()
1124 const KernelDef def;
1256 void OpKernelRegistrar::InitInternal(const KernelDef* kernel_def, in InitInternal()
1411 const KernelDef** def, string* kernel_class_name) { in FindKernelDef()
1446 const KernelDef** def, string* kernel_class_name) { in FindKernelDef()
1532 return GetFilteredRegisteredKernels([](const KernelDef& k) { return true; }); in GetAllRegisteredKernels()
1536 const std::function<bool(const KernelDef&)>& predicate) { in GetFilteredRegisteredKernels()
1542 const KernelDef& kernel_def = p.second.def; in GetFilteredRegisteredKernels()
1551 auto op_pred = [op_name](const KernelDef& k) { return k.op() == op_name; }; in GetRegisteredKernelsForOp()
1688 const KernelDef& kernel_def(key_registration.second.def); in ValidateKernelRegistrations()
|
| D | op_kernel.h | 1456 << ([](::tensorflow::KernelDef const* kernel_def) { \
1503 const KernelDef** def, std::string* kernel_class_name);
1509 const KernelDef** def, std::string* kernel_class_name);
1520 const std::function<bool(const KernelDef&)>& predicate);
1541 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
1548 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
1564 void InitInternal(const KernelDef* kernel_def, StringPiece kernel_class_name,
|
| D | kernel_def_util.cc | 36 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs, in KernelAttrsMatch()
|
| D | memory_types.cc | 91 const KernelDef* kdef = nullptr; in MemoryTypesForNode()
|
| D | op_kernel_test.cc | 579 const KernelDef* kernel_def = nullptr;
1112 auto has_name_test1 = [](const KernelDef& k) { return k.op() == "Test1"; }; in TEST()
1133 auto has_name_test1 = [](const KernelDef& k) { return k.op() == "Test1"; }; in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_supported_ops.cc | 38 std::vector<const KernelDef*> kdefs = in PrintSupportedOps()
43 [](const KernelDef* a, const KernelDef* b) { return a->op() < b->op(); }); in PrintSupportedOps()
48 for (const KernelDef* kdef : kdefs) { in PrintSupportedOps()
50 for (const KernelDef::AttrConstraint& constraint : kdef->constraint()) { in PrintSupportedOps()
|
| D | xla_op_registry.cc | 264 std::unique_ptr<KernelDef> kdef(new KernelDef); in RegisterCompilationKernels()
275 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); in RegisterCompilationKernels()
332 new KernelDef(*kdef), "XlaJitOp", op_registration->factory)); in RegisterCompilationKernels()
339 std::vector<const KernelDef*> XlaOpRegistry::DeviceKernels( in DeviceKernels()
344 std::vector<const KernelDef*> kernels; in DeviceKernels()
350 for (const std::unique_ptr<KernelDef>& k : it->second.kernel_defs) { in DeviceKernels()
|
| D | xla_op_registry.h | 139 typedef bool (*BackendOpFilter)(KernelDef* kdef);
176 static std::vector<const KernelDef*> DeviceKernels(
237 std::vector<std::unique_ptr<KernelDef>> kernel_defs;
|
| D | xla_cpu_backend.cc | 22 bool CpuOpFilter(KernelDef* kdef) { in CpuOpFilter()
|
| D | xla_gpu_backend.cc | 22 bool GpuOpFilter(KernelDef* kdef) { in GpuOpFilter()
|
| D | tf2xla_util.h | 60 KernelDef* kdef);
|
| D | tf2xla_util.cc | 528 KernelDef* kdef) { in AddDtypeToKernelDefConstraint()
529 for (KernelDef::AttrConstraint& constraint : *kdef->mutable_constraint()) { in AddDtypeToKernelDefConstraint()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla_op_registry.clif | 1 from "third_party/tensorflow/core/framework/kernel_def_pyclif.h" import * # KernelDef
7 list<KernelDef>
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | allowlisted_flex_ops_test.cc | 28 auto is_cpu_kernel = [](const tensorflow::KernelDef& def) { in GetAllCpuKernels()
37 tensorflow::KernelDef kernel_def = kernel_list.kernel(i); in GetAllCpuKernels()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 74 const NodeDef& node, const KernelDef** kdef) { in TryFindKernelDef()
76 const KernelDef* kernel = nullptr; in TryFindKernelDef()
161 const KernelDef* kernel = nullptr; in IsNodeOutputPortHostFriendly()
198 const KernelDef* kernel = nullptr; in IsNodeInputPortHostFriendly()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_node_device_util.h | 26 bool TpuOpFilter(KernelDef* kdef);
|
| D | tpu_node_device_util.cc | 24 bool TpuOpFilter(KernelDef* kdef) { in TpuOpFilter()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 581 for (const KernelDef* jit_def : XlaOpRegistry::DeviceKernels( in RegisterXlaDeviceKernels()
584 KernelDef* def = new KernelDef(*jit_def); in RegisterXlaDeviceKernels()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 307 const KernelDef* kdef; in CreateCopyNode()
477 const KernelDef* kdef; in CreateDebugNode()
|