Searched refs:new_output_shape (Results 1 – 5 of 5) sorted by relevance
111 Shape new_output_shape = instruction->shape(); in ExpandInstruction() local113 new_output_shape.DeleteDimension(dim); in ExpandInstruction()170 new_output_shape, {reshaped_input, reshaped_kernel})); in ExpandInstruction()
207 Shape* new_output_shape; in TryResolvePaddedShapesForTensorCore() local208 std::tie(new_input_shape, new_filter_shape, new_output_shape) = [&] { in TryResolvePaddedShapesForTensorCore()229 new_output_shape->dimensions(dnums.output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()240 pad_dim(new_output_shape, dnums.output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
463 auto new_output_shape = in ValidateAndGetTiledExecuteOutputShape() local488 new_output_shape[dimension_index] = in ValidateAndGetTiledExecuteOutputShape()493 new_output_shape, cluster_func_output_type.getElementType()); in ValidateAndGetTiledExecuteOutputShape()
2496 TensorShapeProto* new_output_shape = in SimplifyCase() local2500 *new_output_shape = in SimplifyCase()
2698 new_output_shape = array_ops.concat(2700 output = array_ops.reshape(output, new_output_shape)