Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc41 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); op_idx++) { in QuantizeModel() local
Dquantize_weights.cc39 int32_t op_idx; member
55 for (size_t op_idx = 0; op_idx < subgraph->operators.size(); ++op_idx) { in GetTensorConsumers() local
Dsubgraph_quantizer.cc359 TfLiteStatus SubgraphQuantizer::QuantizeOperator(int op_idx) { in QuantizeOperator()
Dquantize_weights_test.cc96 for (size_t op_idx = 0; op_idx < subgraph->operators()->size(); ++op_idx) { in GetProducerOpCode() local
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AMDGPU/
DR600AsmPrinter.cpp57 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.cc211 for (int op_idx = 0; op_idx < subgraph.operators()->Length(); ++op_idx) { in VerifySubGraphConsistency() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc952 miopenStatus_t SetConvolutionArgs(const int op_idx, const float* alpha, in SetConvolutionArgs()
970 miopenStatus_t SetBiasArgs(const int op_idx, const float* alpha, in SetBiasArgs()
988 miopenStatus_t SetBatchNormInferenceArgs(const int op_idx, const float* alpha, in SetBatchNormInferenceArgs()
1011 miopenStatus_t SetBatchNormForwardArgs(const int op_idx, const float* alpha, in SetBatchNormForwardArgs()
1039 miopenStatus_t SetBatchNormBackwardArgs(const int op_idx, const float* alpha, in SetBatchNormBackwardArgs()
1063 miopenStatus_t SetActivationForwardArgs(const int op_idx, const float* alpha, in SetActivationForwardArgs()
1084 miopenStatus_t SetActivationBackwardArgs(const int op_idx, const float* alpha, in SetActivationBackwardArgs()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc647 for (int64 op_idx : alias_user->OperandIndices(alias.instruction())) { in GetAllUsesOfInstructionAtIndex() local
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py525 def reason(op_idx, details): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc474 for (int op_idx = 0; op_idx < ops.size(); ++op_idx) { in RewireSubgraph() local