Home
last modified time | relevance | path

Searched refs:attr_constraint (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_cpu_backend.cc27 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); in CpuOpFilter() local
28 attr_constraint->set_name("dtype"); in CpuOpFilter()
29 attr_constraint->mutable_allowed_values()->mutable_list()->add_type( in CpuOpFilter()
Dxla_op_registry.cc206 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); in RegisterCompilationKernels() local
207 attr_constraint->set_name(type_attr); in RegisterCompilationKernels()
209 attr_constraint->mutable_allowed_values()->mutable_list(); in RegisterCompilationKernels()