Searched defs:update_shape (Results 1 – 12 of 12) sorted by relevance
96 const Shape& update_shape, const IndexGenerator& start_indices_generator, in EmitDynamicUpdateSliceInPlaceImpl()171 Shape update_shape = update_array.GetShape(); in EmitDynamicUpdateSliceInPlace() local201 Shape update_shape = update->shape(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
50 const TensorShape update_shape = ctx->InputShape("update"); in Compile() local
677 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in Run() local
1067 Shape update_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
1511 const Shape& update_shape = hlo->operand(2)->shape(); in HandleScatter() local
2690 const Shape& operand_shape, const Shape& update_shape, in InferDynamicUpdateSliceShape()
2554 Shape update_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
4655 auto update_shape = dus_update->shape(); in HandleDynamicUpdateSlice() local
5523 const Shape update_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
2344 const auto& update_shape = replicate_update->shape(); in HandleDynamicUpdateSlice() local
691 auto update_shape = this->update().getType().cast<RankedTensorType>(); in fold() local
1950 Shape update_shape = in EmitFusion() local