Searched refs:final_shape_dims (Results 1 – 2 of 2) sorted by relevance
1435 std::vector<int64_t> final_shape_dims( in PropagateOnBroadcast() local1439 final_shape_dims[new_batch_dim] = in PropagateOnBroadcast()1441 final_shape_dims[new_space_dim] *= ctrl_.number_of_splits; in PropagateOnBroadcast()1451 broadcast_dims, final_shape_dims); in PropagateOnBroadcast()
2763 std::optional<nvinfer1::Dims> final_shape_dims = std::nullopt; in ConvertStridedSlice() local2765 final_shape_dims.emplace(); in ConvertStridedSlice()2769 *final_shape_dims = dims_adap->AsTrtDims(); in ConvertStridedSlice()2773 strides, end, final_shape_dims, 0, in ConvertStridedSlice()