Searched defs:update_shape (Results 1 – 10 of 10) 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
2576 const Shape& operand_shape, const Shape& update_shape, in InferDynamicUpdateSliceShape()
2476 Shape update_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
4435 auto update_shape = dus_update->shape(); in HandleDynamicUpdateSlice() local
5471 const Shape update_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
2374 const auto& update_shape = replicate_update->shape(); in HandleDynamicUpdateSlice() local
2036 Shape update_shape = fusion->fused_expression_root()->operand(1)->shape(); in HandleFusion() local