Home
last modified time | relevance | path

Searched refs:squeeze_dims (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dsqueeze.cc50 const int* squeeze_dims = op_context.params->squeeze_dims; in Prepare() local
63 int current = squeeze_dims[idx] < 0 ? squeeze_dims[idx] + input_num_dims in Prepare()
64 : squeeze_dims[idx]; in Prepare()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Ddilated-conv.mlir111 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x64x64x1xf32>) -> tensor<4x64x64xf32>
121 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
132 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x64x64x1xf32>) -> tensor<4x64x64xf32>
142 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
153 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x?x?x1xf32>) -> tensor<4x?x?xf32>
163 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
174 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x?x?x1xf32>) -> tensor<4x?x?xf32>
184 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
195 %3 = "tf.Squeeze"(%2) {squeeze_dims = [3]} : (tensor<4x64x64x1xf32>) -> tensor<4x64x64xf32>
206 …// CHECK-NEXT: [[SQUEEZE:%.*]] = "tf.Squeeze"([[CONV]]) {squeeze_dims = [3]} : (tensor<1x128x128x1…
[all …]
Dlegalize-tf.mlir51 %0 = "tf.Squeeze"(%arg0) {squeeze_dims = [0]} : (tensor<1x1x10xf32>) -> tensor<1x10xf32>
59 // CHECK: "tfl.squeeze"(%arg0) {squeeze_dims = [0]} : (tensor<1x1x10xf32>) -> tensor<1x10xf32>
60 // CHECK: %1 = "tfl.squeeze"(%arg1) {squeeze_dims = []} : (tensor<?x10xf32>) -> tensor<*xf32>
355 // CHECK: "tfl.squeeze"(%arg0) {squeeze_dims = []} : (tensor<1x2x2xf32>) -> tensor<2x2xf32>
359 %0 = "tf.Squeeze"(%arg0) {squeeze_dims = [1]} : (tensor<2x1x2xf32>) -> tensor<2x2xf32>
363 // CHECK: "tfl.squeeze"(%arg0) {squeeze_dims = [1]} : (tensor<2x1x2xf32>) -> tensor<2x2xf32>
367 %0 = "tf.Squeeze"(%arg0) {squeeze_dims = [0, 2]} : (tensor<1x2x1x2xf32>) -> tensor<2x2xf32>
371 // CHECK: "tfl.squeeze"(%arg0) {squeeze_dims = [0, 2]} : (tensor<1x2x1x2xf32>) -> tensor<2x2xf32>
/external/tensorflow/tensorflow/python/ops/losses/
Dutil.py76 …squeeze_dims = lambda: confusion_matrix.remove_squeezable_dimensions( # pylint: disable=g-long-la… function
80 is_last_dim_1, squeeze_dims, lambda: (y_true, y_pred))
82 math_ops.equal(1, rank_diff), maybe_squeeze_dims, squeeze_dims)
/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.h191 std::vector<int32> squeeze_dims; in SqueezeOp() local
192 OP_REQUIRES_OK(ctx, ctx->GetAttr("squeeze_dims", &squeeze_dims)); in SqueezeOp()
193 squeeze_dims_.insert(squeeze_dims.begin(), squeeze_dims.end()); in SqueezeOp()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Squeeze.pbtxt17 name: "squeeze_dims"
30 `squeeze_dims`.
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc200 std::vector<int32> squeeze_dims; in SqueezeOp() local
201 OP_REQUIRES_OK(ctx, ctx->GetAttr("squeeze_dims", &squeeze_dims)); in SqueezeOp()
202 squeeze_dims_.insert(squeeze_dims.begin(), squeeze_dims.end()); in SqueezeOp()
/external/tensorflow/tensorflow/python/kernel_tests/
Dshape_ops_test.py276 def _compareSqueeze(self, x, squeeze_dims, use_gpu): argument
278 if squeeze_dims:
279 np_ans = np.squeeze(x, axis=tuple(squeeze_dims))
280 tensor = array_ops.squeeze(x, squeeze_dims)
289 def _compareSqueezeAll(self, x, squeeze_dims=None): argument
290 if squeeze_dims is None:
291 squeeze_dims = []
292 self._compareSqueeze(x, squeeze_dims, False)
293 self._compareSqueeze(x, squeeze_dims, True)
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc592 const auto& squeeze_dims = schema_params->squeeze_dims(); in ParseOpData() local
594 sizeof(params->squeeze_dims), squeeze_dims, params->squeeze_dims, in ParseOpData()
596 params->num_squeeze_dims = squeeze_dims->size(); in ParseOpData()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSqueeze.pbtxt16 name: "squeeze_dims"
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h337 int squeeze_dims[8]; member
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2090 std::vector<int32> squeeze_dims; in __anonf6523ebd2f02() local
2091 TF_RETURN_IF_ERROR(c->GetAttr("squeeze_dims", &squeeze_dims)); in __anonf6523ebd2f02()
2092 for (int i = 0; i < squeeze_dims.size(); ++i) { in __anonf6523ebd2f02()
2093 if (squeeze_dims[i] < -input_rank || squeeze_dims[i] >= input_rank) { in __anonf6523ebd2f02()
2098 if (squeeze_dims[i] < 0) { in __anonf6523ebd2f02()
2099 squeeze_dims[i] += input_rank; in __anonf6523ebd2f02()
2108 std::find(squeeze_dims.begin(), squeeze_dims.end(), i) != in __anonf6523ebd2f02()
2109 squeeze_dims.end(); in __anonf6523ebd2f02()
2119 if (squeeze_dims.empty()) { in __anonf6523ebd2f02()
2124 if (is_explicit_match || squeeze_dims.empty()) { in __anonf6523ebd2f02()
Darray_ops_test.cc1063 auto rebuild_node_def = [&op](const std::vector<int32>& squeeze_dims) { in TEST() argument
1066 .Attr("squeeze_dims", squeeze_dims) in TEST()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1098 auto squeeze_dims = builder->CreateVector(op.squeeze_dims); in WriteOptions() local
1099 return ::tflite::CreateSqueezeOptions(*builder, squeeze_dims); in WriteOptions()
1104 op->squeeze_dims.insert(op->squeeze_dims.end(), in ReadOptions()
1105 options.squeeze_dims()->begin(), in ReadOptions()
1106 options.squeeze_dims()->end()); in ReadOptions()
Doperator_test.cc459 op.squeeze_dims = {-2, -3, 4, 1, 4}; in TEST_F()
463 EXPECT_EQ(op.squeeze_dims, output_toco_op->squeeze_dims); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1733 std::vector<int> squeeze_dims; in ProcessSqueezeOperator() local
1735 for (int i : op->squeeze_dims) { in ProcessSqueezeOperator()
1736 squeeze_dims.push_back(i < 0 ? i + input_num_dims : i); in ProcessSqueezeOperator()
1740 (!squeeze_dims.empty() && in ProcessSqueezeOperator()
1741 std::find(squeeze_dims.begin(), squeeze_dims.end(), i) == in ProcessSqueezeOperator()
1742 squeeze_dims.end())) { in ProcessSqueezeOperator()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dispatch.py439 def _ragged_squeeze_v1(input, axis=None, name=None, squeeze_dims=None): # pylint: disable=redefine… argument
441 squeeze_dims)
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_patterns.td166 def : Pat<(TF_SqueezeOp $arg, $squeeze_dims), (TFL_SqueezeOp $arg, $squeeze_dims)>;
Doptimize_patterns.td358 def : Pat<(TFL_SqueezeOp:$squeeze_op $input, $squeeze_dims),
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc1496 AttrValue squeeze_dims; in UpdateSqueezeDims() local
1497 squeeze_dims.mutable_list()->mutable_i()->Reserve(squeeze_dims_size); in UpdateSqueezeDims()
1499 squeeze_dims.mutable_list()->mutable_i()->Add(dim); in UpdateSqueezeDims()
1502 node, kAttrSqueezeDims, squeeze_dims); in UpdateSqueezeDims()
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc1124 const auto& squeeze_dims = GetListAttr(node, "squeeze_dims"); in ConvertSqueezeOperator() local
1125 for (int i = 0; i < squeeze_dims.i_size(); ++i) { in ConvertSqueezeOperator()
1126 op->squeeze_dims.push_back(squeeze_dims.i(i)); in ConvertSqueezeOperator()
Dexport_tensorflow.cc1833 if (!src_op.squeeze_dims.empty()) { in ConvertSqueezeOperator()
1834 auto& squeeze_dims = (*new_op->mutable_attr())["squeeze_dims"]; in ConvertSqueezeOperator() local
1835 for (int i : src_op.squeeze_dims) { in ConvertSqueezeOperator()
1836 squeeze_dims.mutable_list()->add_i(i); in ConvertSqueezeOperator()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2037 squeeze_dims = pfor_input.get_attr("squeeze_dims")
2038 squeeze_dims = [i + 1 if i >= 0 else i for i in squeeze_dims]
2039 return wrap(array_ops.squeeze(t, axis=squeeze_dims), True)
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h1047 int32 squeeze_dims[4]; member
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py4041 def squeeze(input, axis=None, name=None, squeeze_dims=None): argument
4086 squeeze_dims)

12