Searched refs:AddDtypeToKernelDefConstraint (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_cpu_backend.cc | 24 AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef); in CpuOpFilter() 27 AddDtypeToKernelDefConstraint("T", DT_STRING, kdef); in CpuOpFilter()
|
D | xla_gpu_backend.cc | 24 AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef); in GpuOpFilter() 27 AddDtypeToKernelDefConstraint("T", DT_STRING, kdef); in GpuOpFilter()
|
D | tf2xla_util.h | 59 void AddDtypeToKernelDefConstraint(absl::string_view name, DataType dtype,
|
D | tf2xla_util.cc | 527 void AddDtypeToKernelDefConstraint(absl::string_view name, DataType dtype, in AddDtypeToKernelDefConstraint() function
|
/external/tensorflow/tensorflow/core/tpu/ |
D | tpu_node_device_util.cc | 29 AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef); in TpuOpFilter() 32 AddDtypeToKernelDefConstraint("T", DT_STRING, kdef); in TpuOpFilter()
|