| /aosp_15_r20/external/pytorch/torch/_custom_op/ |
| H A D | autograd.py | 18 def autograd_kernel_indirection(custom_op): argument
49 def autograd_not_implemented(custom_op): argument
95 custom_op, argument
|
| H A D | impl.py | 51 def custom_op( function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A 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
|
| H A D | raise_custom_ops.cc | 81 auto custom_op = builder.create<CustomTfOp>(loc, op->getResultTypes(), in runOnOperation() local
|
| H A D | partitioned_topological_sort.cc | 43 if (auto custom_op = dyn_cast<mlir::TFL::CustomOp>(*op)) { in IsFlexDelegate() local
|
| H A D | post_quantize.cc | 351 auto custom_op = llvm::isa<CustomOp>(op); in matchAndRewrite() local
|
| H A D | prepare_quantize_dynamic_range.cc | 145 if (auto custom_op = llvm::dyn_cast_or_null<CustomOp>(op)) { in hasInt8QuantizableOperandAt() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | fake_quant_utils.cc | 32 fn.walk([&](TFL::CustomTfOp custom_op) { in UnwrapTFCustomOps()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _custom_ops.py | 24 def custom_op(qualname, func_or_schema=None): function
|
| /aosp_15_r20/external/pytorch/test/custom_operator/ |
| H A D | op.cpp | 9 torch::List<torch::Tensor> custom_op( in custom_op() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/ |
| H A D | verifier_internal_test.cc | 82 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()
|
| H A D | verifier_test.cc | 104 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/core/api/ |
| H A D | flatbuffer_conversions_test.cc | 135 const Operator* custom_op = in TEST_F() local
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| H A D | tflite_engine.cc | 208 const TfLiteRegistration* FindCustomOp(void* user_data, const char* custom_op, in FindCustomOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | c_api_experimental_test.cc | 139 const TfLiteRegistration* MyFindCustomOp(void*, const char* custom_op, in MyFindCustomOp()
|
| /aosp_15_r20/external/pytorch/torch/_library/ |
| H A D | custom_ops.py | 33 def custom_op( function
|
| /aosp_15_r20/external/executorch/backends/qualcomm/utils/ |
| H A D | utils.py | 641 def implement_op(custom_op, op_name, outputs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/ |
| H A D | delegate_test_util.cc | 152 bool automatic_shape_propagation, bool custom_op, in SimpleDelegate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_external_calls.cpp | 1043 auto custom_op = torch::jit::getOperatorForLiteral(custom_op_schema_literal); in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_utils.py | 5057 def custom_op(opname, symbolic_fn, opset_version): function
|