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