Searched refs:reshape_batch_dim (Results 1 – 1 of 1) sorted by relevance
2128 const int reshape_batch_dim = weights_ptr[0]; in ConvertReshape() local2141 if (reshape_batch_dim == -1) { // Other dims of the shape must be fixed. in ConvertReshape()2146 } else if (reshape_batch_dim != input_batch_dim) { in ConvertReshape()2159 << "\nreshape_batch_dim=" << reshape_batch_dim in ConvertReshape()