Searched defs:dim_shape (Results 1 – 6 of 6) sorted by relevance
109 int32_t dim_shape = std::ceil((end - begin) / static_cast<float>(stride)); in CheckOutputSize() local
127 int32_t dim_shape = std::ceil((end - begin) / static_cast<float>(stride)); in ResizeOutputTensor() local
234 const TensorShape dim_shape = ctx->InputShape("dim"); in Compile() local
237 auto* dim_shape = proto->add_dim(); in ShapeHandleToProto() local
2212 const Shape& dim_shape = ShapeUtil::GetSubshape(hlo->shape(), {i}); in HandlePadToStatic() local
855 const Shape& dim_shape = TypeToShape(pad_to_static.output()[i].getType()); in EmitPadToStaticFromMlir() local