Home
last modified time | relevance | path

Searched defs:pad_output (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dmeta_spmd_expander.cc243 mlir::Value pad_output) { in VerifyPaddedDimensionNotSharded()
274 auto pad_output = op->getResult(0); in ExpandOp() local
296 mlir::Value pad_output; in ComputeLayoutForward() local
316 mlir::Value pad_output; in ComputeLayoutBackward() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
H A Dmerge_padding_with.cc142 auto pad_output = graph->FindOutputs(node->id)[0]; in ApplyToNode() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dimage_resize_ops.cc464 bool pad_output = false; in ResizeUsingDilationAndConvolutionGradOp() local