Searched defs:new_output_shape (Results 1 – 8 of 8) sorted by relevance
47 Shape new_output_shape = *input_aligned_output_shape; in HandleReshape() local
111 Shape new_output_shape = instruction->shape(); in ExpandInstruction() local
413 SmallVector<int64_t, 4> new_output_shape; in matchAndRewrite() local572 SmallVector<int64_t, 4> new_output_shape; in matchAndRewrite() local
209 Shape* new_output_shape; in TryResolvePaddedShapesForTensorCore() local
464 Shape new_output_shape = SplitShapeAtDim( in TryVectorizeConv() local
474 auto new_output_shape = in ValidateAndGetTiledExecuteOutputShape() local
232 const Shape new_output_shape = in PartitionGatherIndexPassthroughPartition() local
2546 TensorShapeProto* new_output_shape = in SimplifyCase() local