Searched refs:dims1 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | kernel_util.cc | 98 int64_t dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
100 int64_t out_dims = std::max(dims1, dims2); in CalculateShapeForBroadcast()
104 int64_t d1 = i >= dims1 ? 1 : SizeOfDimension(input1, dims1 - i - 1); in CalculateShapeForBroadcast()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 172 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
175 if (dims1[d] == dims2[d]) in MostSpecificCompatibleShape()
176 output_tensorshape.Concatenate(dims1[d]); in MostSpecificCompatibleShape()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorDimensions.h | 406 static EIGEN_DEVICE_FUNC inline bool run(Dims1& dims1, Dims2& dims2) {
407 return (array_get<n-1>(dims1) == array_get<n-1>(dims2)) &
408 sizes_match_below_dim<Dims1, Dims2, n-1, n-1>::run(dims1, dims2);
422 EIGEN_DEVICE_FUNC bool dimensions_match(Dims1& dims1, Dims2& dims2) {
423 … Dims2, internal::array_size<Dims1>::value, internal::array_size<Dims2>::value>::run(dims1, dims2);
|
| /external/opencv/cv/src/ |
| D | cvhistogram.cpp | 409 int i, dims1, dims2; in cvCompareHist() local
419 CV_CALL( dims1 = cvGetDims( hist1->bins, size1 )); in cvCompareHist()
422 if( dims1 != dims2 ) in cvCompareHist()
426 for( i = 0; i < dims1; i++ ) in cvCompareHist()
659 int i, dims1, dims2; in cvCopyHist() local
674 CV_CALL( dims1 = cvGetDims( src->bins, size1 )); in cvCopyHist()
675 for( i = 0; i < dims1; i++ ) in cvCopyHist()
682 if( dims1 == dims2 ) in cvCopyHist()
684 for( i = 0; i < dims1; i++ ) in cvCopyHist()
689 eq = i == dims1; in cvCopyHist()
[all …]
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contraction.cpp | 329 Eigen::array<DimPair, 2> dims1 = {{DimPair(0, 4), DimPair(1, 0)}}; in test_consistency() local
332 mat3 = mat1.contract(mat2, dims1); in test_consistency()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2118 const int64_t dims1[] = {1, 2}; in TEST_F() local
2119 const size_t ndims1 = TF_ARRAYSIZE(dims1); in TEST_F()
2127 Int8Tensor(dims1, ndims1, tensor1), in TEST_F()
2146 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 220 inline bool AreSameDims(const Dims<4>& dims1, const Dims<4>& dims2) { in AreSameDims() argument
222 if (dims1.sizes[i] != dims2.sizes[i]) { in AreSameDims()
|