Searched defs:update_shape (Results 1 – 7 of 7) sorted by relevance
96 const Shape& update_shape, const IndexGenerator& start_indices_generator, in EmitDynamicUpdateSliceInPlaceImpl()171 Shape update_shape = update_array.GetShape(); in EmitDynamicUpdateSliceInPlace() local204 Shape update_shape = update->shape(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
41 const TensorShape update_shape = ctx->InputShape("update"); in Compile() local
652 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in Run() local
2484 const Shape& operand_shape, const Shape& update_shape, in InferDynamicUpdateSliceShape()
2453 Shape update_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
5024 const Shape update_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
476 Shape update_shape = root->operand(1)->shape(); in HandleFusion() local