Searched refs:output_dynamic_dimension (Results 1 – 2 of 2) sorted by relevance
1063 int64_t output_dynamic_dimension = -1; in HandleReshape() local1069 output_dynamic_dimension = 0; in HandleReshape()1071 output_dynamic_dimension = reshape->shape().rank() - 1; in HandleReshape()1074 if (output_dynamic_dimension == -1) { in HandleReshape()1082 if (output_dynamic_dimension == -1 && in HandleReshape()1085 output_dynamic_dimension = output_dim_start; in HandleReshape()1088 if (output_dynamic_dimension == -1 && in HandleReshape()1127 output_dynamic_dimension = reshape->inferred_dimension(); in HandleReshape()1128 if (output_dynamic_dimension == -1) { in HandleReshape()1132 output_dynamic_dimension = i; in HandleReshape()[all …]
3184 int64_t output_dynamic_dimension = -1; in InferReshapeShape() local3190 output_dynamic_dimension = 0; in InferReshapeShape()3193 output_dynamic_dimension = new_sizes.size() - 1; in InferReshapeShape()3196 if (output_dynamic_dimension == -1) { in InferReshapeShape()3204 if (output_dynamic_dimension == -1 && in InferReshapeShape()3207 output_dynamic_dimension = output_dim_start; in InferReshapeShape()3209 if (output_dynamic_dimension == -1 && in InferReshapeShape()3212 output_dynamic_dimension = inferred_dimension; in InferReshapeShape()3215 if (output_dynamic_dimension != -1) { in InferReshapeShape()3217 inferred_shape.set_dynamic_dimension(output_dynamic_dimension, true); in InferReshapeShape()