Searched defs:dim_shape (Results 1 – 8 of 8) sorted by relevance
169 const int dim_shape = dim_shape_and_index.value(); in ComputeLayoutBackward() local
166 const auto& dim_shape = std::get<SymIntSmallVec>(shape_); in shape_as_dim_vector() 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