Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dsame_shape_propagation.cc316 struct PropagateShapeKnowledgeToKernels struct
318 PropagateShapeKnowledgeToKernels> {
381 return std::make_unique<PropagateShapeKnowledgeToKernels>(); in CreatePropagateShapeKnowledgeToKernels()
H A Dpasses.td108 def PropagateShapeKnowledgeToKernels