| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model.cc | 41 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in QuantizeModel() local
42 auto status = quantizer.QuantizeOperator(op_idx); in QuantizeModel()
44 OperatorT* op = subgraph->operators[op_idx].get(); in QuantizeModel()
49 EnumNameBuiltinOperator(op_code), subgraph_idx, op_idx); in QuantizeModel()
|
| D | quantize_weights.cc | 39 int32_t op_idx; member
55 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); ++op_idx) { in GetTensorConsumers() local
56 OperatorT* op = subgraph->operators[op_idx].get(); in GetTensorConsumers()
63 {op, static_cast<int>(op_idx), static_cast<int>(i)}); in GetTensorConsumers()
335 min_op_idx = std::min(dequant_op_info.op_idx, min_op_idx); in QuantizeWeightsInternal()
|
| D | quantize_weights_test.cc | 96 for (size_t op_idx = 0; op_idx < subgraph->operators()->size(); ++op_idx) { in GetProducerOpCode() local
97 const auto op = subgraph->operators()->Get(op_idx); in GetProducerOpCode()
|
| D | subgraph_quantizer.h | 38 TfLiteStatus QuantizeOperator(int op_idx);
|
| D | subgraph_quantizer.cc | 359 TfLiteStatus SubgraphQuantizer::QuantizeOperator(int op_idx) { in QuantizeOperator() argument
360 OperatorT* op = subgraph_->operators[op_idx].get(); in QuantizeOperator()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 211 for (int op_idx = 0; op_idx < subgraph.operators()->Length(); ++op_idx) { in VerifySubGraphConsistency() local
212 const auto* op = subgraph.operators()->Get(op_idx); in VerifySubGraphConsistency()
224 op_idx, EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
234 op_idx, EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
239 op_idx, EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
245 output_idx, op_idx, in VerifySubGraphConsistency()
252 output_idx, op_idx, in VerifySubGraphConsistency()
|
| /external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AMDGPU/ |
| D | R600AsmPrinter.cpp | 57 for (unsigned op_idx = 0; op_idx < numOperands; op_idx++) { in EmitProgramInfoR600() local
58 const MachineOperand &MO = MI.getOperand(op_idx); in EmitProgramInfoR600()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 123 } else if (optional<int64> op_idx = in AddBackendConstraints() local
125 const HloInstruction* op = instruction->operand(*op_idx); in AddBackendConstraints()
127 ColMajorShape(op->shape()), instruction, *op_idx)); in AddBackendConstraints()
|
| /external/llvm/lib/Target/AMDGPU/ |
| D | AMDGPUAsmPrinter.cpp | 262 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/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 474 for (int op_idx = 0; op_idx < ops.size(); ++op_idx) { in RewireSubgraph() local
475 NodeDef* old_op = ops[op_idx]; in RewireSubgraph()
515 *n->mutable_input(i) = strings::StrCat(sas_name, ":", op_idx); in RewireSubgraph()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 952 miopenStatus_t SetConvolutionArgs(const int op_idx, const float* alpha, in SetConvolutionArgs() argument
955 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &conv_op); in SetConvolutionArgs()
970 miopenStatus_t SetBiasArgs(const int op_idx, const float* alpha, in SetBiasArgs() argument
973 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &bias_op); in SetBiasArgs()
988 miopenStatus_t SetBatchNormInferenceArgs(const int op_idx, const float* alpha, in SetBatchNormInferenceArgs() argument
995 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormInferenceArgs()
1011 miopenStatus_t SetBatchNormForwardArgs(const int op_idx, const float* alpha, in SetBatchNormForwardArgs() argument
1020 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormForwardArgs()
1039 miopenStatus_t SetBatchNormBackwardArgs(const int op_idx, const float* alpha, in SetBatchNormBackwardArgs() argument
1047 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormBackwardArgs()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 647 for (int64 op_idx : alias_user->OperandIndices(alias.instruction())) { in GetAllUsesOfInstructionAtIndex() local
648 uses.emplace_back(alias_user, op_idx); in GetAllUsesOfInstructionAtIndex()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 525 def reason(op_idx, details): argument
528 return '%d %s'%(op_idx, details)
|