| /external/pytorch/torch/autograd/_functions/ |
| D | utils.py | 38 def check_onnx_broadcast(dims1, dims2): argument
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_trace.cpp | 56 array<ptrdiff_t, 2> dims1 = { { 0, 2 } }; in test_simple_trace() local
|
| D | cxx11_tensor_contraction.cpp | 329 Eigen::array<DimPair, 2> dims1 = {{DimPair(0, 4), DimPair(1, 0)}}; in test_consistency() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 176 auto* dims1 = padding_config.add_dimensions(); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 460 const int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
492 const int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
|
| D | kernel_util_test.cc | 339 TfLiteIntArray* dims1 = TfLiteIntArrayCreate(1); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 245 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Linear.cpp | 769 Tensor tensordot(const Tensor& input1, const Tensor& input2, IntArrayRef dims1, IntArrayRef dims2) { in tensordot()
827 Tensor &tensordot_out(const Tensor& input1, const Tensor& input2, IntArrayRef dims1, IntArrayRef di… in tensordot_out()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 307 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel_test.cc | 493 TfLiteIntArray* dims1 = TfLiteIntArrayCreate(1); in TEST() local
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| D | Blas.cpp | 394 IntArrayRef dims1, in tensordot_out()
|
| /external/XNNPACK/test/ |
| D | workspace.cc | 241 std::array<size_t, 4> dims1 = {2, 20, 20, 3}; in TEST() local
|
| /external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| D | constraint_generator.py | 1065 def add_linear_constraints(dims1, dims2, in_features, out_features): argument
|
| /external/pytorch/test/ |
| D | test_utils.py | 801 def try_check_onnx_broadcast(dims1, dims2, expect_broadcast, expect_fail): argument
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2396 const int64_t dims1[] = {1, 2}; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 638 std::vector<int64_t> dims1({1, 2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 130 inline bool AreSameDims(const Dims<4>& dims1, const Dims<4>& dims2) { in AreSameDims()
|