Home
last modified time | relevance | path

Searched refs:Squeeze (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Ddilated-conv.mlir154 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x64x64x1xf32>) -> tensor<4x64x64xf32>
164 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
177 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x64x64x1xf32>) -> tensor<4x64x64xf32>
187 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
200 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x?x?x1xf32>) -> tensor<4x?x?xf32>
210 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
223 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x?x?x1xf32>) -> tensor<4x?x?xf32>
233 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
246 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x64x64x1xf32>) -> tensor<4x64x64xf32>
257 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
[all …]
Dlegalize-tf.mlir53 %0 = "tf.Squeeze"(%arg0) {squeeze_dims = [0]} : (tensor<1x1x10xf32>) -> tensor<1x10xf32>
54 %1 = "tf.Squeeze"(%arg1) : (tensor<?x10xf32>) -> tensor<*xf32>
364 %0 = "tf.Squeeze"(%arg0) : (tensor<1x2x2xf32>) -> tensor<2x2xf32>
372 %0 = "tf.Squeeze"(%arg0) {squeeze_dims = [1]} : (tensor<2x1x2xf32>) -> tensor<2x2xf32>
380 %0 = "tf.Squeeze"(%arg0) {squeeze_dims = [0, 2]} : (tensor<1x2x1x2xf32>) -> tensor<2x2xf32>
Dfuse-tftext.mlir273 %269 = "tf.Squeeze"(%268) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
296 %292 = "tf.Squeeze"(%291) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
451 %447 = "tf.Squeeze"(%446) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
474 %470 = "tf.Squeeze"(%469) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
596 %592 = "tf.Squeeze"(%591) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
619 %615 = "tf.Squeeze"(%614) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
1392 %358 = "tf.Squeeze"(%357) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
1415 %381 = "tf.Squeeze"(%380) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
1574 %540 = "tf.Squeeze"(%539) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
1597 %563 = "tf.Squeeze"(%562) {device = "", squeeze_dims = [1]} : (tensor<?x1xi64>) -> tensor<?xi64>
[all …]
Dprepare-tf.mlir535 …%0 = "tf.Squeeze"(%arg0) {T = f32, _output_shapes = ["tfshape$dim { size: 4 } dim { size: 64 } dim…
706 // CHECK: %[[SQE:.*]] = "tf.Squeeze"(%[[RFF]]) {squeeze_dims = [-2]} : (tensor<10x20x1x30xcomplex<…
/external/tensorflow/tensorflow/examples/wav_to_spectrogram/
Dwav_to_spectrogram.cc71 Output squeeze = Squeeze(root.WithOpName("squeeze"), expand_dims, in WavToSpectrogram()
72 Squeeze::Attrs().Axis({0})); in WavToSpectrogram()
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_Squeeze.pbtxt2 graph_op_name: "Squeeze"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_Squeeze.pbtxt2 graph_op_name: "Squeeze"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSqueeze.pbtxt2 name: "Squeeze"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSqueeze.pbtxt2 name: "Squeeze"
/external/tensorflow/tensorflow/compiler/jit/tests/
Dkeras_imagenet_main_graph_mode.golden_summary47 Squeeze 1
Dkeras_imagenet_main.golden_summary45 Squeeze 1
Dopens2s_gnmt_mixed_precision.golden_summary202 Squeeze 1
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Squeeze.pbtxt2 graph_op_name: "Squeeze"
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc103 auto squeeze_result = Squeeze(scope, matmul_result, Squeeze::Axis(axis)); in SoftmaxCrossEntropyWithLogitsGrad()
Darray_grad_test.cc196 auto y = Squeeze(scope_, x); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc2049 auto squeeze_op = ops::Squeeze( in TEST_F()
2104 auto squeeze_op = ops::Squeeze( in TEST_F()
2144 ops::Squeeze(scope.WithOpName("squeeze").WithDevice("/device:GPU:0"), in TEST_F()
2145 conv2d, ops::Squeeze::Attrs().Axis({1})); in TEST_F()
2184 ops::Squeeze(scope.WithOpName("squeeze").WithDevice("/device:GPU:0"), in TEST_F()
2185 conv2d, ops::Squeeze::Attrs().Axis({1, 2, 3})); in TEST_F()
2224 ops::Squeeze(scope.WithOpName("squeeze").WithDevice("/device:GPU:0"), in TEST_F()
2225 conv2d, ops::Squeeze::Attrs().Axis({1, 2})); in TEST_F()
2287 ops::Squeeze(scope.WithOpName("squeeze").WithDevice("/device:GPU:0"), in TEST_F()
2288 conv2d, ops::Squeeze::Attrs().Axis({-3, -2})); in TEST_F()
[all …]
Dremapper_test.cc953 auto squeeze = ops::Squeeze(s.WithOpName("squeeze"), conv, in TEST_F()
954 ops::Squeeze::Attrs().Axis({2})); in TEST_F()
/external/tensorflow/tensorflow/examples/label_image/
Dmain.cc148 Squeeze(root.WithOpName("squeeze_first_dim"), in ReadTensorFromImageFile()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md529 * `raw_ops.Squeeze`
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_space_to_depth_pass.mlir132 %14 = "tf.Squeeze"(%arg1) {squeeze_dims = [-1]} : (tensor<2x1xf32>) -> tensor<2xf32>
206 %14 = "tf.Squeeze"(%arg1) {squeeze_dims = [-1]} : (tensor<2x1xf32>) -> tensor<2xf32>
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md229 `Squeeze` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
Dcpu_supported_ops.md232 `Squeeze` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1160 class Squeeze class
1916 ops.push_back(MakeUnique<Squeeze>(::tflite::BuiltinOperator_SQUEEZE, in BuildOperatorList()
Doperator_test.cc471 TEST_F(OperatorTest, Squeeze) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc3500 ops::Squeeze::Attrs squeeze_attrs; in TEST_P()
3503 ops::Squeeze(s.WithOpName("my_squeeze"), input, squeeze_attrs); in TEST_P()
3506 auto squeeze = ops::Squeeze(s.WithOpName("my_squeeze"), input); in TEST_P()

12