Home
last modified time | relevance | path

Searched refs:ExpandDims (Results 1 – 25 of 66) sorted by relevance

123

/external/armnn/src/armnnUtils/test/
DTensorUtilsTest.cpp22 armnn::TensorShape outputShape = ExpandDims(inputShape, 0);
35 armnn::TensorShape outputShape = ExpandDims(inputShape, 1);
48 armnn::TensorShape outputShape = ExpandDims(inputShape, 2);
61 armnn::TensorShape outputShape = ExpandDims(inputShape, 3);
74 armnn::TensorShape outputShape = ExpandDims(inputShape, -1);
87 armnn::TensorShape outputShape = ExpandDims(inputShape, -2);
100 armnn::TensorShape outputShape = ExpandDims(inputShape, -3);
113 armnn::TensorShape outputShape = ExpandDims(inputShape, -4);
126 CHECK_THROWS_AS(ExpandDims(inputShape, 4), armnn::InvalidArgumentException);
134 CHECK_THROWS_AS(ExpandDims(inputShape, -5), armnn::InvalidArgumentException);
/external/tensorflow/tensorflow/compiler/jit/tests/
Dopens2s_gnmt_mixed_precision.golden_summary21 ExpandDims 9
71 ExpandDims 2
91 ExpandDims 3
116 ExpandDims 1
146 ExpandDims 2
189 ExpandDims 3
278 ExpandDims 1
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Ddilated-conv.mlir155 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x68x68xf32>, tensor<i32>) -> tensor<4x68x68x1xf32>
165 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
178 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x68x68xf32>, tensor<i32>) -> tensor<4x68x68x1xf32>
188 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
201 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x?x?xf32>, tensor<i32>) -> tensor<4x?x?x1xf32>
211 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
224 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x?x?xf32>, tensor<i32>) -> tensor<4x?x?x1xf32>
234 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
248 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x68x68xf32>, tensor<i32>) -> tensor<4x68x68x1xf32>
259 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
[all …]
Dlower-static-tensor-list-enable-dynamic-update-slice.mlir15 // CHECK-DAG: [[VECTOR_RANK:%.*]] = "tf.ExpandDims"([[ITEM_RANK]], [[ZERO]]) : (tensor<i32>, tens…
18 // CHECK-DAG: [[EXPANDED_INDIX:%.*]] = "tf.ExpandDims"([[INDEX]], [[ZERO]]) : (tensor<i32>, tenso…
20 // CHECK-DAG: [[EXPANDED_ITEM:%.*]] = "tf.ExpandDims"([[ITEM]], [[ZERO]]) : (tensor<10xf32>, tens…
Dlower-static-tensor-list.mlir84 // CHECK: [[VECTOR_RANK:%.*]] = "tf.ExpandDims"([[RANK]], [[ZERO]]) : (tensor<i32>, tensor<i32>) -…
91 // CHECK: [[EXPANDED_START_INDEX:%.*]] = "tf.ExpandDims"([[ZERO]], [[ZERO_2]]) : (tensor<i32>, ten…
93 // CHECK: [[LEADING_DIM:%.*]] = "tf.ExpandDims"([[INDEX]], [[ZERO_2]]) : (tensor<i32>, tensor<i32>…
103 // CHECK: [[EXPANDED_START_INDEX_1:%.*]] = "tf.ExpandDims"([[SUFFIX_START]], [[ZERO_4]]) : (tensor…
105 // CHECK: [[LEADING_DIM_1:%.*]] = "tf.ExpandDims"([[NEG_ONE]], [[ZERO_4]]) : (tensor<i32>, tensor<…
112 // CHECK: [[EXPANDED_ITEM:%.*]] = "tf.ExpandDims"([[ITEM]], [[ZERO]]) : (tensor<10xf32>, tensor<i3…
131 // CHECK: [[VECTOR_RANK:%.*]] = "tf.ExpandDims"([[RANK]], [[ZERO]]) : (tensor<i32>, tensor<i32>) -…
138 // CHECK: [[EXPANDED_START_INDEX:%.*]] = "tf.ExpandDims"([[ZERO]], [[ZERO_2]]) : (tensor<i32>, ten…
140 // CHECK: [[LEADING_DIM:%.*]] = "tf.ExpandDims"([[INDEX]], [[ZERO_2]]) : (tensor<i32>, tensor<i32>…
150 // CHECK: [[EXPANDED_START_INDEX_1:%.*]] = "tf.ExpandDims"([[SUFFIX_START]], [[ZERO_4]]) : (tensor…
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/analysis/
Dcost_analysis.mlir48 …%36 = "tf.ExpandDims"(%28, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
50 …%37 = "tf.ExpandDims"(%29, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
52 …%38 = "tf.ExpandDims"(%30, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
54 …%39 = "tf.ExpandDims"(%31, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
56 …%40 = "tf.ExpandDims"(%32, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
58 …%41 = "tf.ExpandDims"(%33, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
60 …%42 = "tf.ExpandDims"(%34, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
62 …%43 = "tf.ExpandDims"(%35, %11) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tenso…
/external/tensorflow/tensorflow/core/transforms/constant_folding/tests/
Dtrivial_pack.mlir11 … // CHECK: ExpandDims(%[[RANDOM]], %[[CONST_3]]) [%[[CTRL_2]]] name("stack") {T = f32, Tdim = i32}
14 // CHECK: ExpandDims(%[[RANDOM]], %[[CONST_6]]) name("stack_no_axis") {T = f32, Tdim = i32}
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_ExpandDims.pbtxt2 graph_op_name: "ExpandDims"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_ExpandDims.pbtxt2 graph_op_name: "ExpandDims"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlower_tf.mlir39 …// CHECK: %[[INP0:.*]] = "tf.ExpandDims"(%[[ARG0]], %[[AXIS]]) : (tensor<3x5xf32>, tensor<i64>) ->…
40 …// CHECK: %[[INP1:.*]] = "tf.ExpandDims"(%[[ARG1]], %[[AXIS]]) : (tensor<3x5xf32>, tensor<i64>) ->…
170 …// CHECK: %[[INP0:.*]] = "tf.ExpandDims"(%[[ARG0]], %[[AXIS]]) : (tensor<?x5xf32>, tensor<i64>) ->…
171 …// CHECK: %[[INP1:.*]] = "tf.ExpandDims"(%[[ARG1]], %[[AXIS]]) : (tensor<*xf32>, tensor<i64>) -> t…
533 …// CHECK-DAG: %[[RESHAPE:.*]] = "tf.ExpandDims"(%[[ZERO_OR_NAN]], %[[NEG_ONE]]) : (tensor<2xf32>, …
670 // CHECK-DAG: %[[ITEMS_1:.*]] = "tf.ExpandDims"(%[[ITEMS]]#1, %[[AXIS]])
671 // CHECK-DAG: %[[ITEMS_0:.*]] = "tf.ExpandDims"(%[[ITEMS]]#0, %[[AXIS]])
688 // CHECK-DAG: %[[ITEMS1_3:.*]] = "tf.ExpandDims"(%[[ITEMS1]]#3, %[[AXIS]])
689 // CHECK-DAG: %[[ITEMS1_2:.*]] = "tf.ExpandDims"(%[[ITEMS1]]#2, %[[AXIS]])
690 // CHECK-DAG: %[[ITEMS1_1:.*]] = "tf.ExpandDims"(%[[ITEMS1]]#1, %[[AXIS]])
[all …]
Dlayout_optimization_move_transposes_begin.mlir74 …%0 = "tf.ExpandDims"(%arg0, %cst) {device = ""} : (tensor<8x64xf32>, tensor<i32>) -> tensor<8x64x1…
84 …// CHECK: %[[EXPAND_DIMS:.*]] = "tf.ExpandDims"(%arg0, %[[CST_1]]) {device = ""} : (tensor<8x64xf3…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DExpandDims.pbtxt2 name: "ExpandDims"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DExpandDims.pbtxt2 name: "ExpandDims"
/external/tensorflow/tensorflow/c/experimental/ops/
Dupdate_cpp_ops.sh38 ExpandDims \
Darray_ops.h51 Status ExpandDims(AbstractContext* ctx, AbstractTensorHandle* const input,
Darray_ops.cc153 Status ExpandDims(AbstractContext* ctx, AbstractTensorHandle* const input, in ExpandDims() function
/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dexample_proto_fast_parsing_fuzz.cc35 auto in_expanded = ExpandDims(scope, input, Const<int>(scope, 0)); in BuildGraph()
/external/armnn/include/armnnUtils/
DTensorUtils.hpp49 armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis);
/external/tensorflow/tensorflow/examples/wav_to_spectrogram/
Dwav_to_spectrogram.cc70 ExpandDims(root.WithOpName("expand_dims"), cast, expand_dims_const); in WavToSpectrogram()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc650 ExpandDims(scope, Sub(scope, indices_ndims, dim), 0)))}, in GetBatchIndices()
662 auto indices_size = ExpandDims(scope, Size(scope, indices), 0); in BatchGatherGrad()
711 auto indices_size = ExpandDims(scope, Size(scope, indices), 0); in GatherV2Grad()
713 auto axis_expand = ExpandDims(scope, axis, 0); in GatherV2Grad()
738 Output axis_dims_expand = ExpandDims(scope, axis_dims, 0); in GatherV2Grad()
Dnn_grad.cc72 auto reshaped = ExpandDims(scope, vec, -1); in BroadcastMul()
98 auto grad_grad_expand = ExpandDims(scope, grad_grad, 1); in SoftmaxCrossEntropyWithLogitsGrad()
99 auto logits_softmax_expand = ExpandDims(scope, logits_softmax, 2); in SoftmaxCrossEntropyWithLogitsGrad()
/external/tensorflow/tensorflow/compiler/mlir/tfr/tests/
Dend2end.mlir66 // CHECK-NEXT: %[[ED:.*]] = "tf.ExpandDims"(%arg0, %[[AXIS]]) : (tensor<2x3xf32>, tensor<i32>) -> t…
77 // CHECK-NEXT: %[[ED0:.*]] = "tf.ExpandDims"(%arg0, %[[AXIS]]) : (tensor<2x3xf32>, tensor<i32>) -> …
78 // CHECK-NEXT: %[[ED1:.*]] = "tf.ExpandDims"(%arg1, %[[AXIS]]) : (tensor<2x3xf32>, tensor<i32>) -> …
80 // CHECK-NEXT: %[[ED2:.*]] = "tf.ExpandDims"(%arg2, %[[AXIS]]) : (tensor<2x3xf32>, tensor<i32>) -> …
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExpandDims.pbtxt2 graph_op_name: "ExpandDims"
/external/tensorflow/tensorflow/c/experimental/gradients/
Dnn_grad.cc85 ops::ExpandDims(ctx, vec, dim.get(), &expand_dims_outputs, "ExpandDims")); in BroadcastMul()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/
Dtf_cpurt_clustering_policy.mlir525 // tf.ExpandDims
533 %0 = "tf.ExpandDims"(%arg0, %arg1)
543 %0 = "tf.ExpandDims"(%arg0, %arg1)
554 %0 = "tf.ExpandDims"(%arg0, %arg1)
564 %0 = "tf.ExpandDims"(%arg0, %arg1)

123