Searched defs:op_sig (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | gpu_compatibility.cc | 29 const std::string GetOpName(const OpSignature& op_sig) { in GetOpName()
54 absl::Status RetrieveBuiltinData(const OpSignature& op_sig, in RetrieveBuiltinData()
64 absl::Status RetrieveCustomInitialData(const OpSignature& op_sig, in RetrieveCustomInitialData()
93 int GetNumberOfRuntimeInputs(const OpSignature& op_sig) { in GetNumberOfRuntimeInputs()
106 absl::Status CheckInputsOutputs(const OpSignature& op_sig, in CheckInputsOutputs()
129 absl::Status CheckInputsConstsOutputs(const OpSignature& op_sig, in CheckInputsConstsOutputs()
148 absl::Status CheckTensorIsAvailable(const OpSignature& op_sig, int idx) { in CheckTensorIsAvailable()
161 absl::Status CheckConvoultionInputOutput(const OpSignature& op_sig) { in CheckConvoultionInputOutput()
222 absl::Status CheckAxesAreInt32Const(const OpSignature& op_sig, int idx) { in CheckAxesAreInt32Const()
236 absl::Status CheckPooling2DGpuDelegateCompatibility(const OpSignature& op_sig) { in CheckPooling2DGpuDelegateCompatibility()
[all …]
|
| H A D | op_version.cc | 35 bool NeedBroadcastForBinaryInputs(const OpSignature& op_sig) { in NeedBroadcastForBinaryInputs()
42 int GetInputMaxDims(const OpSignature& op_sig) { in GetInputMaxDims()
54 int GetBuiltinOperatorVersion(const OpSignature& op_sig) { in GetBuiltinOperatorVersion()
869 OpSignature op_sig = GetOpSignature(op_code, op, subgraph, model); in UpdateOpVersion() local
|
| H A D | op_signature_test.cc | 148 OpSignature op_sig = GetOpSignature(op_code1, op1, subgraph, model); in TEST() local
208 OpSignature op_sig = in TEST() local
|
| H A D | op_signature.cc | 127 OpSignature op_sig = {builtin_code}; in GetOpSignature() local
257 OpSignature op_sig = { in GetOpSignature() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/ |
| H A D | TunableOp.h | 53 auto op_sig = Signature(); in operator() local
120 auto op_sig = Signature(); in FindFastest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/common/ |
| H A D | delegate_compatibility_checker_base.cc | 100 OpSignature op_sig = tflite::GetOpSignature(op_code, op, subgraph, model); in checkOpCompatibilityOffline() local
|