Searched defs:dims2 (Results 1 – 11 of 11) sorted by relevance
| /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
297 Eigen::array<DimPair, 2> dims2 = {{DimPair(0, 2), DimPair(2, 0)}}; in test_out_of_order_contraction() local
330 Eigen::array<DimPair, 2> dims2 = {{DimPair(4, 0), DimPair(0, 1)}}; in test_consistency() local
|
| D | cxx11_tensor_trace.cpp | 71 array<ptrdiff_t, 2> dims2 = { { 2, 3 } }; in test_simple_trace() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 99 nvinfer1::Dims dims2{1, {2}}; in NetworkWith2Inputs() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 461 const int dims2 = NumDimensions(input2); in CalculateShapeForBroadcast() local
493 const int dims2 = NumDimensions(input2); in CalculateShapeForBroadcast() local
|
| D | kernel_util_test.cc | 344 TfLiteIntArray* dims2 = TfLiteIntArrayCreate(2); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 246 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 308 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel_test.cc | 498 TfLiteIntArray* dims2 = TfLiteIntArrayCreate(2); in TEST() local
|
| /external/XNNPACK/test/ |
| D | workspace.cc | 256 std::array<size_t, 4> dims2 = dims1; in TEST() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2400 const int64_t dims2[] = {2, 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()
|