HomeSort by relevance Sort by last modified time
    Searched refs:kdef (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_gpu_backend.cc 21 bool GpuOpFilter(KernelDef* kdef) {
24 if (kdef->op() == "RandomStandardNormal" || kdef->op() == "RandomUniform" ||
25 kdef->op() == "RandomUniformInt" || kdef->op() == "TruncatedNormal") {
xla_cpu_backend.cc 21 bool CpuOpFilter(KernelDef* kdef) {
24 if (kdef->op() == "RandomStandardNormal") {
25 kdef->clear_constraint();
27 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint();
tf2xla_supported_ops.cc 48 for (const KernelDef* kdef : kdefs) {
50 for (const KernelDef::AttrConstraint& constraint : kdef->constraint()) {
59 std::cout << "`" << kdef->op() << "` | "
xla_op_registry.cc 197 std::unique_ptr<KernelDef> kdef(new KernelDef);
198 kdef->set_op(op_registration->name);
199 kdef->set_device_type(backend.first);
206 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint();
242 !backend.second.op_filter(kdef.get())) {
249 new KernelDef(*kdef), "XlaJitOp", op_registration->factory));
250 backend.second.kernel_defs.push_back(std::move(kdef));
xla_op_registry.h 95 typedef bool (*BackendOpFilter)(KernelDef* kdef);
  /external/tensorflow/tensorflow/compiler/jit/
xla_interpreter_device.cc 61 static bool OpFilter(KernelDef* kdef) { return true; }
  /external/tensorflow/tensorflow/core/framework/
memory_types.cc 79 const KernelDef* kdef = nullptr; local
81 FindKernelDef(device_type, ndef, &kdef, nullptr /* kernel_class_name */);
113 const auto& from_proto = kdef->host_memory_arg();
  /external/tensorflow/tensorflow/core/debug/
debug_graph_utils.cc 274 const KernelDef* kdef; local
310 Status s = FindKernelDef(device_type, node_def, &kdef, nullptr);
445 const KernelDef* kdef; local
465 if (!FindKernelDef(DeviceType(device.device_type()), node_def, &kdef, nullptr)
  /external/tensorflow/tensorflow/core/common_runtime/
constant_folding.cc 523 const KernelDef* kdef; local
524 if (!FindKernelDef(device_type, def, &kdef, nullptr).ok()) {

Completed in 131 milliseconds