Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/jit/tests/
Dopens2s_gnmt_mixed_precision.golden_summary23 ExpandDims 8
90 ExpandDims 3
115 ExpandDims 1
143 ExpandDims 4
185 ExpandDims 3
269 ExpandDims 2
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlower-static-tensor-list.mlir71 // CHECK: [[VECTOR_RANK:%.*]] = "tf.ExpandDims"([[RANK]], [[ZERO]]) : (tensor<i32>, tensor<i32>) -…
78 // CHECK: [[EXPANDED_START_INDEX:%.*]] = "tf.ExpandDims"([[ZERO]], [[ZERO_2]]) : (tensor<i32>, ten…
80 // CHECK: [[LEADING_DIM:%.*]] = "tf.ExpandDims"([[INDEX]], [[ZERO_2]]) : (tensor<i32>, tensor<i32>…
90 // CHECK: [[EXPANDED_START_INDEX_1:%.*]] = "tf.ExpandDims"([[SUFFIX_START]], [[ZERO_4]]) : (tensor…
92 // CHECK: [[LEADING_DIM_1:%.*]] = "tf.ExpandDims"([[NEG_ONE]], [[ZERO_4]]) : (tensor<i32>, tensor<…
99 // CHECK: [[EXPANDED_ITEM:%.*]] = "tf.ExpandDims"([[ITEM]], [[ZERO]]) : (tensor<10xf32>, tensor<i3…
116 // CHECK: [[VECTOR_RANK:%.*]] = "tf.ExpandDims"([[RANK]], [[ZERO]]) : (tensor<i32>, tensor<i32>) -…
123 // CHECK: [[EXPANDED_START_INDEX:%.*]] = "tf.ExpandDims"([[ZERO]], [[ZERO_2]]) : (tensor<i32>, ten…
125 // CHECK: [[LEADING_DIM:%.*]] = "tf.ExpandDims"([[INDEX]], [[ZERO_2]]) : (tensor<i32>, tensor<i32>…
135 // CHECK: [[EXPANDED_START_INDEX_1:%.*]] = "tf.ExpandDims"([[SUFFIX_START]], [[ZERO_4]]) : (tensor…
[all …]
Ddilated-conv.mlir109 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x68x68xf32>, tensor<i32>) -> tensor<4x68x68x1xf32>
119 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
130 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x68x68xf32>, tensor<i32>) -> tensor<4x68x68x1xf32>
140 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
151 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x?x?xf32>, tensor<i32>) -> tensor<4x?x?x1xf32>
161 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
172 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x?x?xf32>, tensor<i32>) -> tensor<4x?x?x1xf32>
182 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
193 %1 = "tf.ExpandDims"(%0, %cst_0) : (tensor<4x68x68xf32>, tensor<i32>) -> tensor<4x68x68x1xf32>
204 …// CHECK-NEXT: [[EXPAND:%.*]] = "tf.ExpandDims"([[INPUT]], [[AXIS]]) : (tensor<1x128x128xf32>, ten…
[all …]
Dprepare-composite-functions-tf.mlir6 %1 = "tf.ExpandDims"(%arg1, %0) : (tensor<*xi32>, tensor<i32>) -> tensor<*xi32>
Dlegalize-tf.mlir343 %0 = "tf.ExpandDims"(%arg0, %arg1) : (tensor<2x2xf32>, tensor<i32>) -> tensor<1x2x2xf32>
/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/core/ops/compat/ops_history_v1/
DExpandDims.pbtxt2 name: "ExpandDims"
/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/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/
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()
Darray_grad_test.cc188 auto y = ExpandDims(scope_, x, 0); in TEST_F()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExpandDims.pbtxt2 graph_op_name: "ExpandDims"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlower_tf.mlir31 …// CHECK: %[[INP0:.*]] = "tf.ExpandDims"(%[[ARG0]], %[[AXIS]]) : (tensor<3x5xf32>, tensor<i64>) ->…
32 …// CHECK: %[[INP1:.*]] = "tf.ExpandDims"(%[[ARG1]], %[[AXIS]]) : (tensor<3x5xf32>, tensor<i64>) ->…
124 …// CHECK: %[[INP0:.*]] = "tf.ExpandDims"(%[[ARG0]], %[[AXIS]]) : (tensor<?x5xf32>, tensor<i64>) ->…
125 …// CHECK: %[[INP1:.*]] = "tf.ExpandDims"(%[[ARG1]], %[[AXIS]]) : (tensor<*xf32>, tensor<i64>) -> t…
265 …// CHECK-DAG: %[[RESHAPE:.*]] = "tf.ExpandDims"(%[[ZERO_OR_NAN]], %[[NEG_ONE]]) : (tensor<2xf32>, …
/external/tensorflow/tensorflow/go/
Dexample_inception_inference_test.go210 op.ExpandDims(s,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dcwise_op_vectorizer.cc83 scope, ops::ExpandDims(scope, rank_diff, ops::Const(scope, 0)), in ExpandDimsForBroadcast()
/external/tensorflow/tensorflow/examples/label_image/
Dmain.cc164 auto dims_expander = ExpandDims(root, float_caster, 0); in ReadTensorFromImageFile()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dophint_lstm.pbtxt192 name: "rnn/TFLiteLSTMCellZeroState/ExpandDims/dim"
213 name: "rnn/TFLiteLSTMCellZeroState/ExpandDims"
214 op: "ExpandDims"
216 input: "rnn/TFLiteLSTMCellZeroState/ExpandDims/dim"
278 input: "rnn/TFLiteLSTMCellZeroState/ExpandDims"
362 op: "ExpandDims"
488 name: "rnn/TFLiteLSTMCellZeroState_1/ExpandDims/dim"
509 name: "rnn/TFLiteLSTMCellZeroState_1/ExpandDims"
510 op: "ExpandDims"
512 input: "rnn/TFLiteLSTMCellZeroState_1/ExpandDims/dim"
[all …]
/external/tensorflow/tensorflow/examples/multibox_detector/
Dmain.cc115 auto dims_expander = ExpandDims(root, float_caster, 0); in ReadTensorFromImageFile()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1260 class ExpandDims class
1859 ops.push_back(MakeUnique<ExpandDims>(::tflite::BuiltinOperator_EXPAND_DIMS, in BuildOperatorList()
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dcpu_supported_ops.md75 `ExpandDims` | `Tdim={int32,int64}`<br>`T={bool,complex64,double,float,int…
Dgpu_supported_ops.md75 `ExpandDims` | `Tdim={int32,int64}`<br>`T={bool,complex64,double,float,int…
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc3863 auto expanded_a = ops::ExpandDims(s.WithOpName("expanded_a"), a, {1}); in TEST_F()
3864 auto expanded_b = ops::ExpandDims(s.WithOpName("expanded_b"), b, {1}); in TEST_F()
3865 auto expanded_c = ops::ExpandDims(s.WithOpName("expanded_c"), c, {1}); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc364 HANDLE_OPERATORTYPENAME_CASE(ExpandDims) in OperatorTypeName()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2006 TEST_F(OpTest, ExpandDims) { in TEST_F() argument

12