Home
last modified time | relevance | path

Searched defs:SoftmaxOpClusteringPolicy (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_clustering.cc604 class SoftmaxOpClusteringPolicy : public DefaultClusteringPolicy { class
606 SoftmaxOpClusteringPolicy() in SoftmaxOpClusteringPolicy() function in tensorflow::__anonff6be9570111::SoftmaxOpClusteringPolicy
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_clustering.cc640 class SoftmaxOpClusteringPolicy : public DefaultClusteringPolicy { class
642 SoftmaxOpClusteringPolicy() in SoftmaxOpClusteringPolicy() function in tensorflow::__anon49e627740111::SoftmaxOpClusteringPolicy