Home
last modified time | relevance | path

Searched defs:pad_shape (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dmirror_pad_op.cc68 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
168 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
H A Dpad_op.cc36 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_space_to_depth_pass.cc182 std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, in GetPadOpForConv2DFilter() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc355 Shape pad_shape = lhs.hlo()->shape(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
H A Dspmd_partitioner.cc3353 const Shape& pad_shape = ShapeUtil::GetSubshape( in HandleInfeed() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_padder.cc852 Shape pad_shape = reverse_shape; in RewriteReverse() local
H A Dalgebraic_simplifier.cc3799 auto pad_shape = pad->shape(); in HandlePad() local