Searched refs:Dims3 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles_test.cc | 34 std::vector<TensorShape> DimVecToShapeVec(std::vector<nvinfer1::Dims3> dimvec) { in DimVecToShapeVec()
90 nvinfer1::Dims3& dims) { in DefineNetwork()
127 nvinfer1::Dims3 dims(8, 8, 10); in TEST_F()
150 std::vector<nvinfer1::Dims3> dim_vec(2, dims); in TEST_F()
158 nvinfer1::Dims3 dims(-1, -1, 10); in TEST_F()
162 std::vector<std::vector<nvinfer1::Dims3>> input_profiles{ in TEST_F()
163 {nvinfer1::Dims3(2, 2, 10), nvinfer1::Dims3(2, 2, 10)}, in TEST_F()
164 {nvinfer1::Dims3(3, 3, 10), nvinfer1::Dims3(3, 3, 10)}, in TEST_F()
165 {nvinfer1::Dims3(16, 16, 10), nvinfer1::Dims3(16, 16, 10)}, in TEST_F()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_index_list.cpp | 57 …List<Eigen::type2index<0>, Eigen::type2index<1>, Eigen::type2index<2>, Eigen::type2index<3>> Dims3; in test_type2index_list() typedef
64 …EIGEN_STATIC_ASSERT((internal::indices_statically_known_to_increase<Dims3>() == true), YOU_MADE_A_… in test_type2index_list()
71 …EIGEN_STATIC_ASSERT((internal::are_inner_most_dims<Dims3, 4, ColMajor>::value == true), YOU_MADE_A… in test_type2index_list()
77 …EIGEN_STATIC_ASSERT((internal::are_inner_most_dims<Dims3, 4, RowMajor>::value == true), YOU_MADE_A… in test_type2index_list()
128 const Dims3 reduction_axis3; in test_type2index_list()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3271 const nvinfer1::Dims3 dilation_dhw( in ConvertConv3DHelper()
3294 const nvinfer1::Dims3 stride_dhw(tf_stride[d_index], tf_stride[h_index], in ConvertConv3DHelper()
3303 nvinfer1::Dims3 effective_kernel_size( in ConvertConv3DHelper()
3355 nvinfer1::Dims3 kernel_size_drs(weights.shape_.d[2], // D in ConvertConv3DHelper()
3474 const nvinfer1::Dims3 stride(tf_stride[d_index], tf_stride[h_index], in ConvertPool3D()
3476 const nvinfer1::Dims3 ksize(tf_kernel[d_index], tf_kernel[h_index], in ConvertPool3D()
|