| /external/tensorflow/tensorflow/lite/kernels/ |
| D | squeeze.cc | 48 const int* squeeze_dims = op_context.params->squeeze_dims; in Prepare() local
61 int current = squeeze_dims[idx] < 0 ? squeeze_dims[idx] + input_num_dims in Prepare()
62 : squeeze_dims[idx]; in Prepare()
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | shape.py | 445 squeeze_dims = []
447 squeeze_dims += [-1]
449 squeeze_dims += [1]
450 if squeeze_dims:
451 x = array_ops.squeeze(x, axis=squeeze_dims)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 163 std::vector<int32> squeeze_dims; in SqueezeOp() local
164 OP_REQUIRES_OK(ctx, ctx->GetAttr("squeeze_dims", &squeeze_dims)); in SqueezeOp()
165 squeeze_dims_.insert(squeeze_dims.begin(), squeeze_dims.end()); in SqueezeOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.h | 191 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/python/keras/utils/ |
| D | losses_utils.py | 101 …squeeze_dims = lambda: confusion_matrix.remove_squeezable_dimensions( # pylint: disable=g-long-la… function
105 is_last_dim_1, squeeze_dims, lambda: (y_true, y_pred))
107 math_ops.equal(1, rank_diff), maybe_squeeze_dims, squeeze_dims)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Squeeze.pbtxt | 17 name: "squeeze_dims"
30 `squeeze_dims`.
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 529 const auto& squeeze_dims = schema_params->squeeze_dims(); in ParseOpData() local
531 sizeof(params->squeeze_dims), squeeze_dims, params->squeeze_dims, in ParseOpData()
533 params->num_squeeze_dims = squeeze_dims->Length(); in ParseOpData()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | shape_ops_test.py | 276 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/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 850 squeeze_dims = []
854 squeeze_dims.append(0)
862 squeeze_dims.append(1)
871 if squeeze_dims:
872 result_op = array_ops.squeeze(result_op, squeeze_dims)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1985 std::vector<int32> squeeze_dims; in __anondb9326b23002() local
1986 TF_RETURN_IF_ERROR(c->GetAttr("squeeze_dims", &squeeze_dims)); in __anondb9326b23002()
1987 for (int i = 0; i < squeeze_dims.size(); ++i) { in __anondb9326b23002()
1988 if (squeeze_dims[i] < -input_rank || squeeze_dims[i] >= input_rank) { in __anondb9326b23002()
1993 if (squeeze_dims[i] < 0) { in __anondb9326b23002()
1994 squeeze_dims[i] += input_rank; in __anondb9326b23002()
2003 std::find(squeeze_dims.begin(), squeeze_dims.end(), i) != in __anondb9326b23002()
2004 squeeze_dims.end(); in __anondb9326b23002()
2014 if (squeeze_dims.empty()) { in __anondb9326b23002()
2019 if (is_explicit_match || squeeze_dims.empty()) { in __anondb9326b23002()
|
| D | array_ops_test.cc | 1065 auto rebuild_node_def = [&op](const std::vector<int32>& squeeze_dims) { in TEST() argument
1068 .Attr("squeeze_dims", squeeze_dims) in TEST()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 302 int squeeze_dims[8]; member
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1720 std::vector<int> squeeze_dims; in ProcessSqueezeOperator() local
1722 for (int i : op->squeeze_dims) { in ProcessSqueezeOperator()
1723 squeeze_dims.push_back(i < 0 ? i + input_num_dims : i); in ProcessSqueezeOperator()
1727 (!squeeze_dims.empty() && in ProcessSqueezeOperator()
1728 std::find(squeeze_dims.begin(), squeeze_dims.end(), i) == in ProcessSqueezeOperator()
1729 squeeze_dims.end())) { in ProcessSqueezeOperator()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 447 op.squeeze_dims = {-2, -3, 4, 1, 4}; in TEST_F()
451 EXPECT_EQ(op.squeeze_dims, output_toco_op->squeeze_dims); in TEST_F()
|
| D | operator.cc | 1407 auto squeeze_dims = builder->CreateVector(op.squeeze_dims); in WriteOptions() local
1408 return ::tflite::CreateSqueezeOptions(*builder, squeeze_dims); in WriteOptions()
1413 op->squeeze_dims.insert(op->squeeze_dims.end(), in ReadOptions()
1414 options.squeeze_dims()->begin(), in ReadOptions()
1415 options.squeeze_dims()->end()); in ReadOptions()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 913 const auto& squeeze_dims = GetListAttr(node, "squeeze_dims"); in ConvertSqueezeOperator() local
914 for (int i = 0; i < squeeze_dims.i_size(); ++i) { in ConvertSqueezeOperator()
915 op->squeeze_dims.push_back(squeeze_dims.i(i)); in ConvertSqueezeOperator()
|
| D | export_tensorflow.cc | 1826 if (!src_op.squeeze_dims.empty()) { in ConvertSqueezeOperator()
1827 auto& squeeze_dims = (*new_op->mutable_attr())["squeeze_dims"]; in ConvertSqueezeOperator() local
1828 for (int i : src_op.squeeze_dims) { in ConvertSqueezeOperator()
1829 squeeze_dims.mutable_list()->add_i(i); in ConvertSqueezeOperator()
|
| D | model.h | 1147 std::vector<int> squeeze_dims;
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | nnapi_delegate.cc | 429 add_vector_int32(builtin->squeeze_dims, in AddOpsAndParams()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3129 def squeeze(input, axis=None, name=None, squeeze_dims=None): argument
3170 "axis", axis, "squeeze_dims", squeeze_dims)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 982 int32 squeeze_dims[4]; member
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 937 int32 squeeze_dims[4]; member
|
| /external/tensorflow/tensorflow/lite/ |
| D | nnapi_delegate.cc | 399 add_vector_int32(builtin->squeeze_dims, in AddOpsAndParams()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_generated.h | 5198 std::vector<int32_t> squeeze_dims;
5208 const flatbuffers::Vector<int32_t> *squeeze_dims() const {
5214 verifier.VerifyVector(squeeze_dims()) &&
5225 void add_squeeze_dims(flatbuffers::Offset<flatbuffers::Vector<int32_t>> squeeze_dims) {
5226 fbb_.AddOffset(SqueezeOptions::VT_SQUEEZE_DIMS, squeeze_dims);
5242 flatbuffers::Offset<flatbuffers::Vector<int32_t>> squeeze_dims = 0) {
5244 builder_.add_squeeze_dims(squeeze_dims);
5250 const std::vector<int32_t> *squeeze_dims = nullptr) {
5253 squeeze_dims ? _fbb.CreateVector<int32_t>(*squeeze_dims) : 0);
9948 … _e = squeeze_dims(); if (_e) { _o->squeeze_dims.resize(_e->size()); for (flatbuffers::uoffset_t _…
[all …]
|
| D | schema.fbs | 571 squeeze_dims:[int];
|