Home
last modified time | relevance | path

Searched defs:op_idx (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc56 int subgraph_index, int op_idx, const string& operator_name, in GetOperatorProperty()
92 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in PopulateRealValueOpSet() local
226 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in InputQuantizeRequired() local
446 for (int op_idx = subgraph->operators.size() - 1; op_idx >= 0; op_idx--) { in ApplyConstraints() local
528 for (int op_idx = subgraph->operators.size() - 1; op_idx >= 0; op_idx--) { in SetOperatorPropertyADDSUBOperator() local
596 ModelT* model, int32_t subgraph_idx, size_t* op_idx, in QuantizeOpInput()
786 ModelT* model, int32_t subgraph_idx, int32_t op_idx, in QuantizeOpOutput()
890 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in QuantizeIntemediateTensors() local
963 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in QuantizeSharedRange() local
1026 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in QuantizeWeightsInputOutput() local
[all …]
Dquantization_wrapper_utils.cc51 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in IntermediateTensorExists() local
89 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in AddIntermediateTensorsToFusedOp() local
Dmodify_model_interface.cc65 for (int32_t op_idx = subgraph->operators.size() - 1; op_idx >= 0; in GetInputTensors() local
140 for (int32_t op_idx = subgraph->operators.size() - 1; op_idx >= 0; in GetOutputTensors() local
Dmodel_utils.cc137 for (int op_idx = subgraph->operators.size() - 1; op_idx >= 0; op_idx--) { in SetOperatorCodeVersion() local
Dquantize_weights.cc42 int32_t op_idx; member
64 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); ++op_idx) { in GetTensorConsumers() local
Dquantize_weights_test.cc120 for (size_t op_idx = 0; op_idx < subgraph->operators()->size(); ++op_idx) { in GetProducerOpCode() local
Dquantization_utils.cc714 float GetEffectiveScale(ModelT* model, SubGraphT* subgraph, int op_idx, in GetEffectiveScale()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dfix_storage_class.cpp33 [&uses](Instruction* use, uint32_t op_idx) { in Process()
175 uint32_t op_idx, std::set<uint32_t>* seen) { in PropagateType()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dfix_storage_class.cpp33 [&uses](Instruction* use, uint32_t op_idx) { in Process()
175 uint32_t op_idx, std::set<uint32_t>* seen) { in PropagateType()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dfix_storage_class.cpp33 [&uses](Instruction* use, uint32_t op_idx) { in Process()
175 uint32_t op_idx, std::set<uint32_t>* seen) { in PropagateType()
/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
/external/llvm-project/llvm/lib/Target/AMDGPU/
DR600AsmPrinter.cpp56 for (unsigned op_idx = 0; op_idx < numOperands; op_idx++) { in EmitProgramInfoR600() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc123 } else if (optional<int64> op_idx = in AddBackendConstraints() local
/external/llvm/lib/Target/AMDGPU/
DAMDGPUAsmPrinter.cpp262 for (unsigned op_idx = 0; op_idx < numOperands; op_idx++) { in EmitProgramInfoR600() local
332 for (unsigned op_idx = 0; op_idx < numOperands; op_idx++) { in getSIProgramInfo() local
/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc527 for (int op_idx = 0, end = subgraph.operators()->size(); op_idx < end; in VerifySubGraphConsistency() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation.cc1205 int64 op_idx = user.operand_index(&instruction); in GetShardingFromUser() local
1304 int64 op_idx = user.operand_index(&instruction); in GetShardingFromUser() local
Dtuple_points_to_analysis.cc726 for (int64 op_idx : alias_user->OperandIndices(alias.instruction())) { in GetAllUsesOfInstructionAtIndex() local
Dhlo_rematerialization.cc475 for (int64 op_idx : user->OperandIndices(buffer_alias.instruction())) { in GetUsers() local
1568 const int64 op_idx = user.operand_number; in RematerializeInstructions() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1069 miopenStatus_t SetConvolutionArgs(const int op_idx, const float* alpha, in SetConvolutionArgs()
1087 miopenStatus_t SetBiasArgs(const int op_idx, const float* alpha, in SetBiasArgs()
1105 miopenStatus_t SetBatchNormInferenceArgs(const int op_idx, const float* alpha, in SetBatchNormInferenceArgs()
1129 const int op_idx, const float* alpha, const float* beta, in SetBatchNormForwardArgs()
1152 miopenStatus_t SetBatchNormBackwardArgs(const int op_idx, const float* alpha, in SetBatchNormBackwardArgs()
1176 miopenStatus_t SetActivationForwardArgs(const int op_idx, const float* alpha, in SetActivationForwardArgs()
1197 miopenStatus_t SetActivationBackwardArgs(const int op_idx, const float* alpha, in SetActivationBackwardArgs()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2993 for (int op_idx : llvm::seq<int>(0, old_num_operands)) { in matchAndRewrite() local
3033 for (int op_idx : llvm::seq<int>(0, old_num_operands)) { in matchAndRewrite() local
3059 for (int op_idx : llvm::seq<int>(0, old_num_operands)) { in matchAndRewrite() local
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py501 def reason(op_idx, details): argument
/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