Searched refs:dims3 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel_test.cc | 404 TfLiteIntArray* dims3 = TfLiteIntArrayCreate(3); in TEST() local
405 dims3->data[0] = 4; in TEST()
406 dims3->data[1] = 5; in TEST()
407 dims3->data[2] = 6; in TEST()
408 EXPECT_EQ("[4,5,6]", GetDimsDebugString(dims3)); in TEST()
409 free(dims3); in TEST()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_convolution.cpp | 28 Eigen::array<Tensor<float, 2>::Index, 1> dims3{{0}}; in test_evals() local
30 typedef TensorEvaluator<decltype(input.convolve(kernel, dims3)), DefaultDevice> Evaluator; in test_evals()
31 Evaluator eval(input.convolve(kernel, dims3), DefaultDevice()); in test_evals()
|
| D | cxx11_tensor_contraction.cpp | 32 Eigen::array<DimPair, 1> dims3 = {{DimPair(0, 0)}}; in test_evals() local
33 typedef TensorEvaluator<decltype(mat1.contract(mat2, dims3)), DefaultDevice> Evaluator; in test_evals()
34 Evaluator eval(mat1.contract(mat2, dims3), DefaultDevice()); in test_evals()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 441 int dims3 = NumDimensions(input3); in CalculateShapeForBroadcast() local
442 int out_dims = std::max(std::max(dims1, dims2), dims3); in CalculateShapeForBroadcast()
448 int d3 = i >= dims3 ? 1 : SizeOfDimension(input3, dims3 - i - 1); in CalculateShapeForBroadcast()
|