Searched defs:dim_shape (Results 1 – 7 of 7) sorted by relevance
169 const int dim_shape = dim_shape_and_index.value(); in ComputeLayoutBackward() local
214 int32_t dim_shape = std::ceil((end - begin) / static_cast<float>(stride)); in ResizeOutputTensor() local
262 const TensorShape dim_shape = ctx->InputShape("dim"); in Compile() local
646 auto dim_shape = Concat( in GetBatchIndices() local
244 auto* dim_shape = proto->add_dim(); in ShapeHandleToProto() local
2341 const Shape& dim_shape = ShapeUtil::GetSubshape(hlo->shape(), {i}); in HandlePadToStatic() local
772 const Shape& dim_shape = GetShape(pad_to_static.getOutput()[i]); in EmitPadToStatic() local