Home
last modified time | relevance | path

Searched refs:kSqueeze (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_squeeze_attributes.cc33 if (squeeze_op->type != OperatorType::kSqueeze) { in Run()
Dconvert_squeeze_to_reshape.cc38 if (squeeze_it->get()->type != OperatorType::kSqueeze) { in Run()
Dreorder_elementwise_unary.cc54 case OperatorType::kSqueeze: in IsMoveOperator()
Didentify_dilated_conv.cc73 if (post_conv_op->type != OperatorType::kSqueeze) { in ResolveDilatedConv()
Dhardcode_min_max.cc437 case OperatorType::kSqueeze: in Run()
Dquantize.cc50 type == OperatorType::kSqueeze || type == OperatorType::kPad || in SupportsQuantization()
Dpropagate_fixed_sizes.cc2221 case OperatorType::kSqueeze: in Run()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h96 kSqueeze, enumerator
1145 SqueezeOperator() : Operator(OperatorType::kSqueeze) {}
Dexport_tensorflow.cc2268 } else if (src_op.type == OperatorType::kSqueeze) { in ConvertOperator()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc49 constexpr char kSqueeze[] = "Squeeze"; variable
267 {kSqueeze, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc450 GetOperator("SQUEEZE", OperatorType::kSqueeze), op); in TEST_F()
Doperator.cc2418 OperatorType::kSqueeze)); in BuildOperatorList()