Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_padding_mapper.cc42 constexpr char kReplicatedInputIndicesAttr[] = "_replicated_input_indices"; variable
191 replicate->getAttrOfType<ArrayAttr>(kReplicatedInputIndicesAttr); in RemapAndAssignPaddingMaps()
Dtpu_cluster_formation.cc57 constexpr char kReplicatedInputIndicesAttr[] = "_replicated_input_indices"; variable
429 replicate_op->setAttr(kReplicatedInputIndicesAttr, in ReplicateCluster()