Searched refs:operator_property (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_quantize_helper.h | 81 namespace operator_property = ::tflite::optimize::operator_property; variable
87 LstmOp op, operator_property::OpVariant* lstm_variant, in GetLstmProperty()
88 operator_property::OperatorProperty* op_property) { in GetLstmProperty()
105 *op_property = operator_property::GetOperatorProperty(*lstm_variant); in GetLstmProperty()
117 [&](std::pair<int, operator_property::TensorProperty> input) { in GetLstmProperty()
124 [&](std::pair<int, operator_property::TensorProperty> in GetLstmProperty()
140 operator_property::OpVariant lstm_variant; in matchAndRewrite()
141 operator_property::OperatorProperty lstm_property; in matchAndRewrite()
235 SourceOp op, const operator_property::OpVariant& op_variant, in processInputs()
236 const operator_property::OperatorProperty& op_property, in processInputs()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model.cc | 54 operator_property::OperatorProperty GetOperatorProperty( in GetOperatorProperty()
58 operator_property::OperatorProperty property = in GetOperatorProperty()
59 operator_property::GetOperatorProperty(model, subgraph_index, op_idx); in GetOperatorProperty()
95 operator_property::OperatorProperty property = in PopulateRealValueOpSet()
104 for (const std::pair<int, operator_property::TensorProperty>& input : in PopulateRealValueOpSet()
113 for (const std::pair<int, operator_property::TensorProperty>& output : in PopulateRealValueOpSet()
449 operator_property::OperatorProperty property = in ApplyConstraints()
551 std::vector<std::pair<int, operator_property::TensorProperty>> GetInputs( in GetInputs()
552 const OperatorT* op, operator_property::OperatorProperty property) { in GetInputs()
553 std::vector<std::pair<int, operator_property::TensorProperty>> inputs; in GetInputs()
[all …]
|
| D | BUILD | 64 ":operator_property",
106 ":operator_property",
159 ":operator_property",
191 name = "operator_property",
192 srcs = ["operator_property.cc"],
193 hdrs = ["operator_property.h"],
301 ":operator_property",
|
| D | quantization_wrapper_utils.cc | 92 operator_property::OperatorProperty property = in AddIntermediateTensorsToFusedOp()
93 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in AddIntermediateTensorsToFusedOp()
|
| D | model_utils.cc | 140 operator_property::OperatorProperty property = in SetOperatorCodeVersion()
141 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in SetOperatorCodeVersion()
|
| D | operator_property.h | 23 namespace operator_property {
|
| D | operator_property.cc | 22 namespace operator_property { namespace
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | BUILD | 569 "//tensorflow/lite/tools/optimize:operator_property",
|