Searched defs:pad_shape (Results 1 – 7 of 7) sorted by relevance
68 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local168 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
36 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
181 std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, in GetPadOpForConv2DFilter() local
357 Shape pad_shape = lhs.hlo()->shape(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
2572 const Shape& pad_shape = in HandleInfeed() local
797 Shape pad_shape = reverse_shape; in RewriteReverse() local
3510 auto pad_shape = pad->shape(); in HandlePad() local