Searched refs:update_or_gather_shape (Results 1 – 1 of 1) sorted by relevance
801 const Shape& update_or_gather_shape, in PassthroughOperandToGatherOutputOrScatterUpdate() argument809 std::vector<int64> passthrough_tile(update_or_gather_shape.rank(), 1); in PassthroughOperandToGatherOutputOrScatterUpdate()