Searched refs:dims1 (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 211 int64_t dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
213 int64_t out_dims = std::max(dims1, dims2); in CalculateShapeForBroadcast()
221 int64_t d1 = i >= dims1 ? 1 : SizeOfDimension(input1, dims1 - i - 1); in CalculateShapeForBroadcast()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 195 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
198 if (dims1[d] == dims2[d]) in MostSpecificCompatibleShape()
199 output_tensorshape.AddDim(dims1[d]); in MostSpecificCompatibleShape()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 263 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
266 if (dims1[d] == dims2[d]) in MostSpecificCompatibleShape()
267 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/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 | 2318 const int64_t dims1[] = {1, 2}; in TEST_F() local
2319 const size_t ndims1 = TF_ARRAYSIZE(dims1); in TEST_F()
2327 Int8Tensor(dims1, ndims1, tensor1), 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/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 388 std::vector<int64> dims1({1, 2}); in TEST_F() local
391 dims1, add_computation)); in TEST_F()
|
| /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()
|