Searched refs:op_reg (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_util_test.cc | 89 auto op_reg = opts.op_registry(); in TEST() local
96 op_reg); in TEST()
101 NodeBuilder stack_op_builder("stack_op", "StackV2", op_reg); in TEST()
105 NodeBuilder stack_close_builder("stack_close", "StackCloseV2", op_reg); in TEST()
124 auto op_reg = opts.op_registry(); in TEST() local
131 op_reg); in TEST()
136 NodeBuilder stack_op_builder("stack_op", "StackV2", op_reg); in TEST()
141 op_reg); in TEST()
145 NodeBuilder stack_close_builder("stack_close", "StackCloseV2", op_reg); in TEST()
164 auto op_reg = opts.op_registry(); in TEST() local
[all …]
|
| D | xla_op_registry.cc | 205 [](const std::unique_ptr<OpRegistration>& op_reg) { in RegisterCompilationKernels() argument
206 return op_reg->has_device_allowlist; in RegisterCompilationKernels()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 407 void* OpInit(const TfLiteRegistration& op_reg, const char* buffer, in OpInit() argument
409 if (op_reg.init == nullptr) return nullptr; in OpInit()
410 return op_reg.init(&context_, buffer, length); in OpInit()
414 void OpFree(const TfLiteRegistration& op_reg, void* buffer) { in OpFree() argument
415 if (op_reg.free == nullptr) return; in OpFree()
417 op_reg.free(&context_, buffer); in OpFree()
422 TfLiteStatus OpPrepare(const TfLiteRegistration& op_reg, TfLiteNode* node);
425 TfLiteStatus OpInvoke(const TfLiteRegistration& op_reg, TfLiteNode* node) { in OpInvoke() argument
426 if (op_reg.invoke == nullptr) return kTfLiteError; in OpInvoke()
427 return op_reg.invoke(&context_, node); in OpInvoke()
|
| D | subgraph.cc | 146 const char* GetTFLiteOpName(const TfLiteRegistration& op_reg) { in GetTFLiteOpName() argument
147 if (op_reg.builtin_code == tflite::BuiltinOperator_CUSTOM) { in GetTFLiteOpName()
148 const char* const custom_name = op_reg.custom_name; in GetTFLiteOpName()
151 if (op_reg.builtin_code == tflite::BuiltinOperator_DELEGATE && in GetTFLiteOpName()
152 op_reg.custom_name) { in GetTFLiteOpName()
153 return op_reg.custom_name; in GetTFLiteOpName()
155 return tflite::EnumNamesBuiltinOperator()[op_reg.builtin_code]; in GetTFLiteOpName()
879 TfLiteStatus Subgraph::OpPrepare(const TfLiteRegistration& op_reg, in OpPrepare() argument
881 if (op_reg.prepare == nullptr) { in OpPrepare()
883 if (IsUnresolvedCustomOp(op_reg)) { in OpPrepare()
[all …]
|
| /external/capstone/contrib/objdump/ |
| D | objdump-m68k.py | 46 def dump_op_reg(insn, op_reg): argument
47 if op_reg == M68K_REG_A7:
49 if op_reg == M68K_REG_A6:
51 return '%' + insn.reg_name(op_reg)
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.h | 589 const char* OpProfilingString(const TfLiteRegistration& op_reg, in OpProfilingString() argument
591 if (op_reg.profiling_string == nullptr) return nullptr; in OpProfilingString()
592 return op_reg.profiling_string(context_, node); in OpProfilingString()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | tf_tfl_flatbuffer_helpers.cc | 169 const OpRegistrationData* op_reg = in RegisterCustomBuiltinOps() local
171 if (op_reg) continue; in RegisterCustomBuiltinOps()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util.cc | 436 const OpRegistrationData* op_reg; in DoesAnyCalleeHaveRefNodes() local
437 if (OpRegistry::Global()->LookUp(call_target.name(), &op_reg).ok()) { in DoesAnyCalleeHaveRefNodes()
438 const OpDef& op = op_reg->op_def; in DoesAnyCalleeHaveRefNodes()
|
| /external/mesa3d/src/amd/compiler/ |
| D | aco_lower_to_hw_instr.cpp | 930 PhysReg op_reg = src.op.physReg(); in split_copy() local
932 op_reg.reg_b += offset; in split_copy()
943 can_increase = op_reg.reg_b % next == 0; in split_copy()
966 *op = Operand(op_reg, op_cls); in split_copy()
|