Searched refs:squeeze_op (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_squeeze_to_reshape.cc | 41 auto squeeze_op = static_cast<SqueezeOperator*>(squeeze_it->get()); in Run() local
42 CHECK_EQ(squeeze_op->inputs.size(), 1); in Run()
43 CHECK_EQ(squeeze_op->outputs.size(), 1); in Run()
45 const auto& input_array = model->GetArray(squeeze_op->inputs[0]); in Run()
54 if (!model->HasArray(squeeze_op->outputs[0]) || in Run()
55 !model->GetArray(squeeze_op->outputs[0]).has_shape()) { in Run()
61 const auto& output_shape = model->GetArray(squeeze_op->outputs[0]).shape(); in Run()
70 squeeze_op->inputs[0], in Run()
71 CreateInt32Array(model, squeeze_op->outputs[0] + "_shape", in Run()
74 reshape_op->outputs = squeeze_op->outputs; in Run()
[all …]
|
| D | resolve_squeeze_attributes.cc | 32 auto* squeeze_op = model->operators[op_index].get(); in Run() local
33 if (squeeze_op->type != OperatorType::kSqueeze) { in Run()
36 DCHECK_EQ(squeeze_op->inputs.size(), 1); in Run()
37 DCHECK_EQ(squeeze_op->outputs.size(), 1); in Run()
40 if (CountOpsWithInput(*model, squeeze_op->outputs[0]) == 1) { in Run()
41 const auto* next_op = GetOpWithInput(*model, squeeze_op->outputs[0]); in Run()
46 LogName(*squeeze_op)); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 113 TF::SqueezeOp squeeze_op; in matchAndRewrite() local
122 squeeze_op = llvm::cast<TF::SqueezeOp>(next_op); in matchAndRewrite()
140 auto squeeze_dims = squeeze_op.squeeze_dims(); in matchAndRewrite()
286 squeeze_op.getResult().setType(bts_op.output().getType()); in matchAndRewrite()
290 biasadd_op.setOperand(0, squeeze_op.output()); in matchAndRewrite()
291 biasadd_op.output().setType(squeeze_op.output().getType()); in matchAndRewrite()
|
| D | optimize_patterns.td | 493 (TFL_SqueezeOp:$squeeze_op $input, $squeeze_dims),
494 (TFL_ReshapeOp $input, (ConstantOp (GetShape $squeeze_op))),
495 [(HasValidRankedTensor $squeeze_op)]>;
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 114 auto squeeze_op = createReshapeOp(slice_op.output(), {num_rows, num_cols}, in sliceInput() local
117 sliced.emplace_back(squeeze_op.output()); in sliceInput()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 2049 auto squeeze_op = ops::Squeeze( in TEST_F() local
2051 auto z = ops::Identity(scope.WithOpName("z"), squeeze_op); in TEST_F()
2104 auto squeeze_op = ops::Squeeze( in TEST_F() local
2106 auto z = ops::Identity(scope.WithOpName("z"), squeeze_op); in TEST_F()
2143 auto squeeze_op = in TEST_F() local
2146 auto z = ops::Identity(scope.WithOpName("z"), squeeze_op); in TEST_F()
2183 auto squeeze_op = in TEST_F() local
2186 auto z = ops::Identity(scope.WithOpName("z"), squeeze_op); in TEST_F()
2223 auto squeeze_op = in TEST_F() local
2226 auto z = ops::Identity(scope.WithOpName("z"), squeeze_op); in TEST_F()
[all …]
|