| /external/tensorflow/tensorflow/core/framework/ |
| D | kernel_def_builder_test.cc | 26 const KernelDef* def = KernelDefBuilder("A").Device(DEVICE_CPU).Build(); in TEST()
27 KernelDef expected; in TEST()
35 const KernelDef* def = KernelDefBuilder("B") in TEST()
39 KernelDef expected; in TEST()
76 const KernelDef* def = KernelDefBuilder("E") in TEST()
81 KernelDef expected; in TEST()
|
| D | kernel_def_builder.h | 27 class KernelDef; variable
73 const KernelDef* Build();
76 KernelDef* kernel_def_;
|
| D | kernel_def_builder.cc | 24 kernel_def_ = new KernelDef; in KernelDefBuilder()
74 const KernelDef* KernelDefBuilder::Build() { in Build()
75 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 | 11 message KernelDef { message
45 repeated KernelDef kernel = 1;
|
| D | kernel_def_util.h | 26 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs,
|
| D | op_kernel.cc | 942 KernelRegistration(const KernelDef& d, StringPiece c, in KernelRegistration()
946 const KernelDef def;
1068 void OpKernelRegistrar::InitInternal(const KernelDef* kernel_def, in InitInternal()
1148 const KernelDef** def, string* kernel_class_name) { in FindKernelDef()
1215 return GetFilteredRegisteredKernels([](const KernelDef& k) { return true; }); in GetAllRegisteredKernels()
1219 const std::function<bool(const KernelDef&)>& predicate) { in GetFilteredRegisteredKernels()
1224 const KernelDef& kernel_def = p.second.def; in GetFilteredRegisteredKernels()
1233 auto op_pred = [op_name](const KernelDef& k) { return k.op() == op_name; }; in GetRegisteredKernelsForOp()
1350 const KernelDef& kernel_def(key_registration.second.def); in ValidateKernelRegistrations()
|
| D | kernel_def_util.cc | 36 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs, in KernelAttrsMatch()
|
| D | op_kernel.h | 1422 const KernelDef** def, string* kernel_class_name);
1433 const std::function<bool(const KernelDef&)>& predicate);
1454 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
1465 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
1485 void InitInternal(const KernelDef* kernel_def, StringPiece kernel_class_name,
|
| D | memory_types.cc | 84 const KernelDef* kdef = nullptr; in MemoryTypesForNode()
|
| D | op_kernel_test.cc | 504 const KernelDef* kernel_def = nullptr;
1003 auto has_name_test1 = [](const KernelDef& k) { return k.op() == "Test1"; }; in TEST()
1024 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 | 253 std::unique_ptr<KernelDef> kdef(new KernelDef); in RegisterCompilationKernels()
263 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); in RegisterCompilationKernels()
317 new KernelDef(*kdef), "XlaJitOp", op_registration->factory)); in RegisterCompilationKernels()
324 std::vector<const KernelDef*> XlaOpRegistry::DeviceKernels( in DeviceKernels()
329 std::vector<const KernelDef*> kernels; in DeviceKernels()
335 for (const std::unique_ptr<KernelDef>& k : it->second.kernel_defs) { in DeviceKernels()
|
| D | xla_op_registry.h | 104 typedef bool (*BackendOpFilter)(KernelDef* kdef);
137 static std::vector<const KernelDef*> DeviceKernels(
193 std::vector<std::unique_ptr<KernelDef>> kernel_defs;
|
| D | xla_gpu_backend.cc | 22 bool GpuOpFilter(KernelDef* kdef) { in GpuOpFilter()
|
| D | xla_cpu_backend.cc | 22 bool CpuOpFilter(KernelDef* kdef) { in CpuOpFilter()
|
| D | tf2xla_util.h | 59 KernelDef* kdef);
|
| D | tf2xla_util.cc | 529 KernelDef* kdef) { in AddDtypeToKernelDefConstraint()
530 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/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 72 const NodeDef& node, const KernelDef** kdef) { in TryFindKernelDef()
74 const KernelDef* kernel = nullptr; in TryFindKernelDef()
157 const KernelDef* kernel = nullptr; in IsNodeOutputPortHostFriendly()
194 const KernelDef* kernel = nullptr; in IsNodeInputPortHostFriendly()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 530 for (const KernelDef* jit_def : XlaOpRegistry::DeviceKernels( in RegisterXlaDeviceKernels()
533 KernelDef* def = new KernelDef(*jit_def); in RegisterXlaDeviceKernels()
|
| D | xla_interpreter_device.cc | 78 static bool OpFilter(KernelDef* kdef) { return true; } in OpFilter()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 274 const KernelDef* kdef; in CreateCopyNode()
444 const KernelDef* kdef; in CreateDebugNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 533 const KernelDef* kdef; in ReplaceTensorWithConstant()
|