Searched refs:dims2 (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 212 int64_t dims2 = NumDimensions(input2); in CalculateShapeForBroadcast() local
213 int64_t out_dims = std::max(dims1, dims2); in CalculateShapeForBroadcast()
222 int64_t d2 = i >= dims2 ? 1 : SizeOfDimension(input2, dims2 - i - 1); in CalculateShapeForBroadcast()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contraction.cpp | 145 Eigen::array<DimPair, 2> dims2({{DimPair(0, 1), DimPair(1, 0)}}); in test_multidims() local
146 typedef TensorEvaluator<decltype(mat4.contract(mat5, dims2)), DefaultDevice> Evaluator2; in test_multidims()
147 Evaluator2 eval2(mat4.contract(mat5, dims2), DefaultDevice()); in test_multidims()
297 Eigen::array<DimPair, 2> dims2 = {{DimPair(0, 2), DimPair(2, 0)}}; in test_out_of_order_contraction() local
298 mat3 = mat1.contract(mat2, dims2); in test_out_of_order_contraction()
330 Eigen::array<DimPair, 2> dims2 = {{DimPair(4, 0), DimPair(0, 1)}}; in test_consistency() local
333 mat4 = mat2.contract(mat1, dims2); in test_consistency()
|
| /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/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 196 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
198 if (dims1[d] == dims2[d]) in MostSpecificCompatibleShape()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 264 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
266 if (dims1[d] == dims2[d]) in MostSpecificCompatibleShape()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2322 const int64_t dims2[] = {2, 2}; in TEST_F() local
2323 const size_t ndims2 = TF_ARRAYSIZE(dims2); in TEST_F()
2328 Int8Tensor(dims2, ndims2, tensor2), in TEST_F()
2346 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 129 inline bool AreSameDims(const Dims<4>& dims1, const Dims<4>& dims2) { in AreSameDims() argument
131 if (dims1.sizes[i] != dims2.sizes[i]) { in AreSameDims()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 273 inline bool AreSameDims(const Dims<4>& dims1, const Dims<4>& dims2) { in AreSameDims() argument
275 if (dims1.sizes[i] != dims2.sizes[i]) { in AreSameDims()
|