Searched defs:pad_shape (Results 1 – 6 of 6) sorted by relevance
68 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local167 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
34 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
227 std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, in GetPadOpForConv2DFilter() local
357 Shape pad_shape = lhs.hlo()->shape(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
2597 const Shape& pad_shape = in HandleInfeed() local
3420 auto pad_shape = pad->shape(); in HandlePad() local