Searched refs:final_shape_dims (Results 1 – 2 of 2) sorted by relevance
1025 std::vector<int64> final_shape_dims( in PropagateOnBroadcast() local1029 final_shape_dims[new_batch_dim] = in PropagateOnBroadcast()1031 final_shape_dims[new_space_dim] *= kNumSplits; in PropagateOnBroadcast()1039 broadcast_dims, final_shape_dims); in PropagateOnBroadcast()
3201 nvinfer1::Dims final_shape_dims; in ConvertStridedSlice() local3204 final_shape_dims = in ConvertStridedSlice()3206 final_shape_dims_ptr = &final_shape_dims; in ConvertStridedSlice()