Home
last modified time | relevance | path

Searched refs:SupportsValueSpecialization (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir/utils/runtime/
Dconstraints.h38 inline bool SupportsValueSpecialization(mlir::Type type) { in SupportsValueSpecialization() function
Dconstraints.cc85 if (SupportsValueSpecialization(shaped)) return constraint; in ResolveArgumentConstraint()
/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
Dspecialization.cc220 if (!tensor || !SupportsValueSpecialization(tensor)) { in SpecializeFunction()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_clustering.cc735 tfrt::cpu::jit::SupportsValueSpecialization(value.getType())) in VerifyCluster()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_clustering.cc906 xla::runtime::SupportsValueSpecialization(value.getType())) in VerifyCluster()