Searched defs:custom_op (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | quantize.cc | 80 auto custom_op = llvm::dyn_cast_or_null<TFL::CustomOp>(op); in IsQuantizableCustomOp() local
115 if (auto custom_op = dyn_cast_or_null<CustomOp>(quantized_op)) { in IsWeightOnlyOp() local
|
| D | raise_custom_ops.cc | 81 auto custom_op = builder.create<CustomTfOp>(loc, op->getResultTypes(), in runOnOperation() local
|
| D | partitioned_topological_sort.cc | 43 if (auto custom_op = dyn_cast<mlir::TFL::CustomOp>(*op)) { in IsFlexDelegate() local
|
| D | post_quantize.cc | 351 auto custom_op = llvm::isa<CustomOp>(op); in matchAndRewrite() local
|
| D | prepare_quantize_dynamic_range.cc | 145 if (auto custom_op = llvm::dyn_cast_or_null<CustomOp>(op)) { in hasInt8QuantizableOperandAt() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | fake_quant_utils.cc | 32 fn.walk([&](TFL::CustomTfOp custom_op) { in UnwrapTFCustomOps()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier_internal_test.cc | 82 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()
|
| D | verifier_test.cc | 104 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | tflite_engine.cc | 208 const TfLiteRegistration* FindCustomOp(void* user_data, const char* custom_op, in FindCustomOp()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions_test.cc | 135 const Operator* custom_op = in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_experimental_test.cc | 139 const TfLiteRegistration* MyFindCustomOp(void*, const char* custom_op, in MyFindCustomOp()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test_util.cc | 152 bool automatic_shape_propagation, bool custom_op, in SimpleDelegate()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1234 if (auto custom_op = dyn_cast<mlir::TFL::CustomOp>(inst)) { in BuildOperator() local
1596 if (auto custom_op = dyn_cast<mlir::TFL::CustomTfOp>(inst)) { in BuildSubGraph() local
|