Searched refs:output_dynamic_dimension (Results 1 – 2 of 2) sorted by relevance
889 int64 output_dynamic_dimension = -1; in HandleReshape() local895 output_dynamic_dimension = 0; in HandleReshape()898 output_dynamic_dimension = reshape->shape().rank() - 1; in HandleReshape()901 if (output_dynamic_dimension == -1) { in HandleReshape()909 if (output_dynamic_dimension == -1 && in HandleReshape()912 output_dynamic_dimension = output_dim_start; in HandleReshape()915 if (output_dynamic_dimension == -1 && in HandleReshape()954 output_dynamic_dimension = reshape->inferred_dimension(); in HandleReshape()955 if (output_dynamic_dimension == -1) { in HandleReshape()959 output_dynamic_dimension = i; in HandleReshape()[all …]
3044 int64 output_dynamic_dimension = -1; in InferReshapeShape() local3050 output_dynamic_dimension = 0; in InferReshapeShape()3053 output_dynamic_dimension = new_sizes.size() - 1; in InferReshapeShape()3056 if (output_dynamic_dimension == -1) { in InferReshapeShape()3064 if (output_dynamic_dimension == -1 && in InferReshapeShape()3067 output_dynamic_dimension = output_dim_start; in InferReshapeShape()3069 if (output_dynamic_dimension == -1 && in InferReshapeShape()3072 output_dynamic_dimension = inferred_dimension; in InferReshapeShape()3075 if (output_dynamic_dimension != -1) { in InferReshapeShape()3077 inferred_shape.set_dynamic_dimension(output_dynamic_dimension, true); in InferReshapeShape()