Searched defs:dim_shape (Results 1 – 6 of 6) sorted by relevance
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
645 auto dim_shape = Concat( in GetBatchIndices() local
244 auto* dim_shape = proto->add_dim(); in ShapeHandleToProto() local
2218 const Shape& dim_shape = ShapeUtil::GetSubshape(hlo->shape(), {i}); in HandlePadToStatic() local
718 const Shape& dim_shape = GetShape(pad_to_static.output()[i]); in EmitPadToStatic() local