Searched refs:custom_op (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | raise_custom_ops.cc | 55 auto custom_op = builder.create<CustomTfOp>( in runOnFunction() local
63 custom_op.body().takeBody(region); in runOnFunction()
65 op->replaceAllUsesWith(custom_op); in runOnFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | raise-custom-ops.mlir | 3 // CHECK-LABEL: custom_op
4 func @custom_op(%arg0: tensor<4xf32>) -> tensor<4xf32> {
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions_test.cc | 135 const Operator* custom_op = in TEST_F() local
139 ParseOpData(custom_op, BuiltinOperator_CUSTOM, &mock_reporter_, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_test.py | 375 custom_op = op_hint.OpHint(name)
376 input1 = custom_op.add_input(input1)
377 input2 = custom_op.add_input(input2)
379 return custom_op.add_output(output)
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_experimental.h | 89 const char* custom_op,
|
| D | c_api_experimental_test.cc | 138 const TfLiteRegistration* MyFindCustomOp(void*, const char* custom_op, in MyFindCustomOp() argument
140 if (absl::string_view(custom_op) == "foo" && version == 1) { in MyFindCustomOp()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | tflite_engine.cc | 222 const TfLiteRegistration* FindCustomOp(void* user_data, const char* custom_op, in FindCustomOp() argument
225 return op_resolver->FindOp(custom_op, version); in FindCustomOp()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test_util.h | 77 bool custom_op = true);
|
| D | delegate_test_util.cc | 133 bool automatic_shape_propagation, bool custom_op) in SimpleDelegate() argument
139 custom_op_(custom_op) { in SimpleDelegate()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier_test.cc | 104 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator() argument
106 CreateOperatorCodeDirect(builder_, builtin_op, custom_op)); in AddOperator()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1125 if (auto custom_op = dyn_cast<mlir::TFL::CustomOp>(inst)) { in BuildOperator() local
1126 return BuildCustomOperator(inst, custom_op, operands, results); in BuildOperator()
1456 if (auto custom_op = dyn_cast<mlir::TFL::CustomTfOp>(inst)) { in BuildSubGraph() local
1459 if (!custom_op.body().empty()) { in BuildSubGraph()
1460 real_inst = &custom_op.body().front().front(); in BuildSubGraph()
|
| /external/tensorflow/tensorflow/ |
| D | opensource_only.files | 290 tensorflow/tools/ci_build/release/ubuntu_16/custom_op/nightly.sh
291 tensorflow/tools/ci_build/release/ubuntu_16/custom_op/release.sh
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | BUILD | 260 "//tensorflow/lite/tools/optimize:testdata/custom_op.bin",
|