Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dmeta_spmd_expander.cc242 mlir::Value pad_input, in VerifyPaddedDimensionNotSharded()
273 auto pad_input = op->getOperand(0); in ExpandOp() local
295 mlir::Value pad_input; in ComputeLayoutForward() local
315 mlir::Value pad_input; in ComputeLayoutBackward() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc558 auto pad_input = pad_op.input(); in HandleFirstConvolution() local