Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_clustering.cc110 class TensorflowOpClusteringPolicy : public ClusteringPolicy { class
207 : public TensorflowOpClusteringPolicy<BroadcastToOp> {
391 : public TensorflowOpClusteringPolicy<ConcatV2Op> {
417 class ConstOpClusteringPolicy : public TensorflowOpClusteringPolicy<ConstOp> {
435 : public TensorflowOpClusteringPolicy<ExpandDimsOp> {
459 : public TensorflowOpClusteringPolicy<_FusedMatMulOp> {
495 class FillOpClusteringPolicy : public TensorflowOpClusteringPolicy<FillOp> {
528 class OneHotOpClusteringPolicy : public TensorflowOpClusteringPolicy<OneHotOp> {
554 class RangeOpClusteringPolicy : public TensorflowOpClusteringPolicy<RangeOp> {
580 : public TensorflowOpClusteringPolicy<ReshapeOp> {
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_clustering.cc105 class TensorflowOpClusteringPolicy : public ClusteringPolicy { class
195 : public TensorflowOpClusteringPolicy<BroadcastToOp> {
378 : public TensorflowOpClusteringPolicy<ConcatV2Op> {
404 class ConstOpClusteringPolicy : public TensorflowOpClusteringPolicy<ConstOp> {
422 : public TensorflowOpClusteringPolicy<ExpandDimsOp> {
446 : public TensorflowOpClusteringPolicy<_FusedMatMulOp> {
482 class FillOpClusteringPolicy : public TensorflowOpClusteringPolicy<FillOp> {
521 class RangeOpClusteringPolicy : public TensorflowOpClusteringPolicy<RangeOp> {
547 : public TensorflowOpClusteringPolicy<ReshapeOp> {
580 class ShapeOpClusteringPolicy : public TensorflowOpClusteringPolicy<ShapeOp> {
[all …]