| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/ |
| D | unsigned_reduction_of_ones_like.mlir | 3 …%dims1 = "tf.Const"() { value = dense<[2]> : tensor<1xi32>, device = "/job:localhost/replica:0/tas…
4 …%1 = "tf.Sum"(%0, %dims1) { keep_dims = false, device = "/job:localhost/replica:0/task:0/device:CP…
|
| D | all_of_approximate_equal_true.mlir | 7 %dims1 = "tf.Const"() {
11 %1 = "tf.All"(%0, %dims1) {
|
| D | reduction_of_broadcast.mlir | 3 …%dims1 = "tf.Const"() { value = dense<[0]> : tensor<1xi32>, device = "/job:localhost/replica:0/tas…
4 …%1 = "tf.Max"(%0, %dims1) {keep_dims = false, device = "/job:localhost/replica:0/task:0/device:CPU…
|
| D | transpose_of_asinh.mlir | 3 …%dims1 = "tf.Const"() { value = dense<[2, 1, 3, 0]> : tensor<4xi32>, device = "/job:localhost/repl…
4 …%1 = "tf.Transpose"(%0, %dims1) { device = "/job:localhost/replica:0/task:0/device:CPU:0" } : (ten…
|
| D | reduction_of_reduction_static.mlir | 4 …%dims1 = "tf.Const"() { value = dense<[1, 2, 3]> : tensor<3xi32>, device = "/job:localhost/replica…
5 …%1 = "tf.Max"(%0, %dims1) { keep_dims = false, device = "/job:localhost/replica:0/task:0/device:CP…
|
| D | reduction_of_reduction.mlir | 4 %dims1 = "tf.Const"() {
8 %1 = "tf.Mean"(%V__0, %dims1) {
|
| D | floor_div_of_prod_and_oneslike.mlir | 3 …%dims1 = "tf.Const"() { value = dense<[2]> : tensor<1xi32>, device = "/job:localhost/replica:0/tas…
4 …%1 = "tf.Prod"(%0, %dims1) { keep_dims = false, device = "/job:localhost/replica:0/task:0/device:C…
|
| D | equal_of_reduction.mlir | 4 …%dims1 = "tf.Const"() { value = dense<[1]> : tensor<1xi32>, device = "/job:localhost/replica:0/tas…
5 …%1 = "tf.Sum"(%0, %dims1) { keep_dims = false, device = "/job:localhost/replica:0/task:0/device:CP…
|
| D | select_of_sum.mlir | 3 …%dims1 = "tf.Const"() { value = dense<[1]> : tensor<1xi32>, device = "/job:localhost/replica:0/tas…
4 …%1 = "tf.Sum"(%0, %dims1) { keep_dims = true, device = "/job:localhost/replica:0/task:0/device:CPU…
|
| D | broadcast_of_reduction_of_reduction.mlir | 2 …%dims1 = "tf.Const"() { value = dense<[0, 1]> : tensor<2xi32>, device = "/job:localhost/replica:0/…
3 …%1 = "tf.Max"(%V__0, %dims1) {keep_dims = true, device = "/job:localhost/replica:0/task:0/device:C…
|
| D | bias_add_of_reduction_ops.mlir | 4 …%dims1 = "tf.Const"() { value = dense<[0, 2]> : tensor<2xi32>, device = "/job:localhost/replica:0/…
5 …%1 = "tf.Sum"(%0, %dims1) {keep_dims = true, device = "/job:localhost/replica:0/task:0/device:CPU:…
|
| D | nested_reductions.mlir | 4 …%dims1 = "tf.Const"() { value = dense<[0, 1]> : tensor<2xi32>, device = "/job:localhost/replica:0/…
5 …%1 = "tf.Max"(%0, %dims1) { keep_dims = true, device = "/job:localhost/replica:0/task:0/device:CPU…
|
| D | sum_of_min_of_min.mlir | 4 …%dims1 = "tf.Const"() { value = dense<[0, 1]> : tensor<2xi32>, device = "/job:localhost/replica:0/…
5 …%1 = "tf.Min"(%0, %dims1) { keep_dims = true, device = "/job:localhost/replica:0/task:0/device:CPU…
|
| D | broadcasting_24.mlir | 3 …%dims1 = "tf.Const"() { value = dense<[1]> : tensor<1xi32>, device = "/job:localhost/replica:0/tas…
4 …%1 = "tf.Min"(%V__1, %dims1) { keep_dims = true, device = "/job:localhost/replica:0/task:0/device:…
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 460 const int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
462 const int out_dims = std::max(dims1, dims2); in CalculateShapeForBroadcast()
467 const int d1 = i >= dims1 ? 1 : SizeOfDimension(input1, dims1 - i - 1); in CalculateShapeForBroadcast()
492 const int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
495 const int out_dims = std::max(std::max(dims1, dims2), dims3); in CalculateShapeForBroadcast()
499 const int d1 = i >= dims1 ? 1 : SizeOfDimension(input1, dims1 - i - 1); in CalculateShapeForBroadcast()
|
| D | kernel_util_test.cc | 339 TfLiteIntArray* dims1 = TfLiteIntArrayCreate(1); in TEST_F() local
340 dims1->data[0] = 1; in TEST_F()
341 EXPECT_EQ("[1]", GetShapeDebugString(dims1)); in TEST_F()
342 TfLiteIntArrayFree(dims1); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 176 auto* dims1 = padding_config.add_dimensions(); in TEST_F() local
177 dims1->set_edge_padding_low(3); in TEST_F()
178 dims1->set_interior_padding(1); in TEST_F()
179 dims1->set_edge_padding_high(0); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel_test.cc | 493 TfLiteIntArray* dims1 = TfLiteIntArrayCreate(1); in TEST() local
494 dims1->data[0] = 1; in TEST()
495 EXPECT_EQ("[1]", GetShapeDebugString(dims1)); in TEST()
496 TfLiteIntArrayFree(dims1); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 245 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
248 if (dims1[d] == dims2[d]) in MostSpecificCompatibleShape()
249 output_tensorshape.AddDim(dims1[d]); in MostSpecificCompatibleShape()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 307 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
310 if (dims1[d] == dims2[d]) in MostSpecificCompatibleShape()
311 output_tensorshape.Concatenate(dims1[d]); in MostSpecificCompatibleShape()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorDimensions.h | 468 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool run(Dims1& dims1, Dims2& dims2) {
469 return (array_get<n-1>(dims1) == array_get<n-1>(dims2)) &&
470 sizes_match_below_dim<Dims1, Dims2, n-1, n-1>::run(dims1, dims2);
484 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE bool dimensions_match(Dims1 dims1, Dims2 dims2) {
485 … Dims2, internal::array_size<Dims1>::value, internal::array_size<Dims2>::value>::run(dims1, dims2);
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_trace.cpp | 56 array<ptrdiff_t, 2> dims1 = { { 0, 2 } }; in test_simple_trace() local
57 Tensor<float, 1, DataLayout> result1 = tensor1.trace(dims1); in test_simple_trace()
|
| 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/XNNPACK/test/ |
| D | workspace.cc | 241 std::array<size_t, 4> dims1 = {2, 20, 20, 3}; in TEST() local
244 DefineGraph(&subgraph1, dims1); in TEST()
256 std::array<size_t, 4> dims2 = dims1; in TEST()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2396 const int64_t dims1[] = {1, 2}; in TEST_F() local
2397 const size_t ndims1 = TF_ARRAYSIZE(dims1); in TEST_F()
2405 Int8Tensor(dims1, ndims1, tensor1), in TEST_F()
2424 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F()
|