Home
last modified time | relevance | path

Searched refs:op_idx (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc53 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in DuplicateBiasesWithMultipleUses() local
55 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in DuplicateBiasesWithMultipleUses()
56 auto* op = subgraph->operators[op_idx].get(); in DuplicateBiasesWithMultipleUses()
79 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in DuplicateBiasesWithMultipleUses() local
81 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in DuplicateBiasesWithMultipleUses()
82 OperatorT* op = subgraph->operators[op_idx].get(); in DuplicateBiasesWithMultipleUses()
148 int subgraph_index, int op_idx, const string& operator_name, in GetOperatorProperty() argument
151 operator_property::GetOperatorProperty(model, subgraph_index, op_idx); in GetOperatorProperty()
153 const OperatorT* op = subgraph->operators[op_idx].get(); in GetOperatorProperty()
196 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in IsConstantWithOneUse() local
[all …]
Dquantization_wrapper_utils.cc53 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in IntermediateTensorExists() local
54 OperatorT* op = subgraph->operators[op_idx].get(); in IntermediateTensorExists()
91 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in AddIntermediateTensorsToFusedOp() local
93 OperatorT* op = subgraph->operators[op_idx].get(); in AddIntermediateTensorsToFusedOp()
95 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in AddIntermediateTensorsToFusedOp()
107 auto name = CreateTensorName(op_idx, i); in AddIntermediateTensorsToFusedOp()
Dmodel_utils.cc138 for (int op_idx = subgraph->operators.size() - 1; op_idx >= 0; op_idx--) { in SetOperatorCodeVersion() local
139 OperatorT* op = subgraph->operators[op_idx].get(); in SetOperatorCodeVersion()
142 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in SetOperatorCodeVersion()
Dmodify_model_interface.cc68 for (int32_t op_idx = subgraph->operators.size() - 1; op_idx >= 0; in GetInputTensors() local
69 op_idx--) { in GetInputTensors()
70 OperatorT* op = subgraph->operators[op_idx].get(); in GetInputTensors()
117 result.push_back({subgraph_idx, op->inputs[0], op_idx, op->outputs[0], in GetInputTensors()
143 for (int32_t op_idx = subgraph->operators.size() - 1; op_idx >= 0; in GetOutputTensors() local
144 op_idx--) { in GetOutputTensors()
145 OperatorT* op = subgraph->operators[op_idx].get(); in GetOutputTensors()
192 result.push_back({subgraph_idx, op->inputs[0], op_idx, op->outputs[0], in GetOutputTensors()
Dquantize_weights_portable.cc42 int32_t op_idx; member
64 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); ++op_idx) { in GetTensorConsumers() local
65 OperatorT* op = subgraph->operators[op_idx].get(); in GetTensorConsumers()
72 {op, static_cast<int32_t>(op_idx), static_cast<int32_t>(i)}); in GetTensorConsumers()
485 min_op_idx = std::min(dequant_op_info.op_idx, min_op_idx);
569 min_op_idx = std::min(dequant_op_info.op_idx, min_op_idx); in QuantizeWeightsFloat16()
Dquantize_weights.cc44 int32_t op_idx; member
78 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); ++op_idx) { in GetTensorConsumers() local
79 OperatorT* op = subgraph->operators[op_idx].get(); in GetTensorConsumers()
86 {op, static_cast<int32_t>(op_idx), static_cast<int32_t>(i)}); in GetTensorConsumers()
499 min_op_idx = std::min(dequant_op_info.op_idx, min_op_idx);
583 min_op_idx = std::min(dequant_op_info.op_idx, min_op_idx); in QuantizeWeightsFloat16()
Dquantize_weights_test.cc120 for (size_t op_idx = 0; op_idx < subgraph->operators()->size(); ++op_idx) { in GetProducerOpCode() local
121 const auto op = subgraph->operators()->Get(op_idx); in GetProducerOpCode()
Dquantization_utils.h145 float GetEffectiveScale(ModelT* model, SubGraphT* subgraph, int op_idx,
/external/tensorflow/tensorflow/lite/experimental/telemetry/
Dprofiler.h48 virtual void ReportTelemetryOpEvent(const char* event_name, int64_t op_idx,
69 virtual uint32_t ReportBeginOpInvokeEvent(const char* op_name, int64_t op_idx,
78 int64_t op_idx, int64_t subgraph_idx) = 0;
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/
DR600AsmPrinter.cpp56 for (unsigned op_idx = 0; op_idx < numOperands; op_idx++) { in EmitProgramInfoR600() local
57 const MachineOperand &MO = MI.getOperand(op_idx); in EmitProgramInfoR600()
/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc523 for (int op_idx = 0, end = subgraph.operators()->size(); op_idx < end; in VerifySubGraphConsistency() local
524 ++op_idx) { in VerifySubGraphConsistency()
525 const auto* op = subgraph.operators()->Get(op_idx); in VerifySubGraphConsistency()
545 input_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
555 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
561 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
567 output_idx, op_idx, in VerifySubGraphConsistency()
574 output_idx, op_idx, in VerifySubGraphConsistency()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dfix_storage_class.cpp33 [&uses](Instruction* use, uint32_t op_idx) { in Process() argument
34 uses.push_back({use, op_idx}); in Process()
175 uint32_t op_idx, std::set<uint32_t>* seen) { in PropagateType() argument
187 if (op_idx == 2) { in PropagateType()
200 if (op_idx > 2) { in PropagateType()
Dfix_storage_class.h79 bool PropagateType(Instruction* inst, uint32_t type_id, uint32_t op_idx,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc124 } else if (optional<int64_t> op_idx = in AddBackendConstraints() local
126 const HloInstruction* op = instruction->operand(*op_idx); in AddBackendConstraints()
128 SetOperandLayout(ColMajorShape(op->shape()), instruction, *op_idx)); in AddBackendConstraints()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dfix_storage_class.cpp33 [&uses](Instruction* use, uint32_t op_idx) { in Process() argument
34 uses.push_back({use, op_idx}); in Process()
175 uint32_t op_idx, std::set<uint32_t>* seen) { in PropagateType() argument
187 if (op_idx == 2) { in PropagateType()
200 if (op_idx > 2) { in PropagateType()
Dfix_storage_class.h79 bool PropagateType(Instruction* inst, uint32_t type_id, uint32_t op_idx,
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dfix_storage_class.cpp33 [&uses](Instruction* use, uint32_t op_idx) { in Process() argument
34 uses.push_back({use, op_idx}); in Process()
175 uint32_t op_idx, std::set<uint32_t>* seen) { in PropagateType() argument
187 if (op_idx == 2) { in PropagateType()
200 if (op_idx > 2) { in PropagateType()
Dfix_storage_class.h79 bool PropagateType(Instruction* inst, uint32_t type_id, uint32_t op_idx,
/external/tensorflow/tensorflow/lite/python/
Dtest_util.py29 for op_idx in range(subgraph.OperatorsLength()):
30 op = subgraph.Operators(op_idx)
/external/llvm/lib/Target/AMDGPU/
DAMDGPUAsmPrinter.cpp262 for (unsigned op_idx = 0; op_idx < numOperands; op_idx++) { in EmitProgramInfoR600() local
263 const MachineOperand &MO = MI.getOperand(op_idx); in EmitProgramInfoR600()
332 for (unsigned op_idx = 0; op_idx < numOperands; op_idx++) { in getSIProgramInfo() local
333 const MachineOperand &MO = MI.getOperand(op_idx); in getSIProgramInfo()
/external/ComputeLibrary/src/core/experimental/
DPostOpUtils.h45 int op_idx = 0; in transform_post_op_list_arguments() local
73 ++op_idx; in transform_post_op_list_arguments()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc3196 for (int op_idx : llvm::seq<int>(0, old_num_operands)) { in matchAndRewrite() local
3197 auto body_arg = body_block.getArgument(op_idx); in matchAndRewrite()
3198 auto yield_operand = LookThroughIdentity(yield.getOperand(op_idx)); in matchAndRewrite()
3199 auto while_operand = while_op.getOperand(op_idx); in matchAndRewrite()
3208 auto cond_arg = cond_block.getArgument(op_idx); in matchAndRewrite()
3212 auto result = while_op.getResult(op_idx); in matchAndRewrite()
3219 if (body_block.getArgument(op_idx).use_empty() && in matchAndRewrite()
3220 cond_block.getArgument(op_idx).use_empty() && in matchAndRewrite()
3221 while_op.getResult(op_idx).use_empty()) { in matchAndRewrite()
3222 removed_operand.set(op_idx); in matchAndRewrite()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dquantize_weights_test.cc129 for (size_t op_idx = 0; op_idx < subgraph->operators()->size(); ++op_idx) { in GetProducerOpCode() local
130 const auto op = subgraph->operators()->Get(op_idx); in GetProducerOpCode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc728 for (int op_idx = 0, idx_limit = ops.size(); op_idx < idx_limit; ++op_idx) { in RewireSubgraph() local
729 NodeDef* old_op = ops[op_idx]; in RewireSubgraph()
769 *n->mutable_input(i) = strings::StrCat(sas_name, ":", op_idx); in RewireSubgraph()
/external/tensorflow/tensorflow/lite/tools/optimize/debugging/python/
Ddebugger.py364op_idx = self._defining_op[tensor_detail['index']] # pytype: disable=unsupported-operands # dyna…
365 … op_detail = self._quant_interpreter._get_op_details(op_idx) # pylint: disable=protected-access

12