Searched refs:n_inputs (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.h | 74 int n_inputs = network->getNbInputs(); in SetDimensions() local
75 if (min.size() != n_inputs || opt.size() != n_inputs || in SetDimensions()
76 max.size() != n_inputs) { in SetDimensions()
79 for (int i = 0; i < n_inputs; i++) { in SetDimensions()
|
| D | trt_shape_optimization_profiles.cc | 189 int n_inputs = network->getNbInputs(); in SetShapeTensorMask() local
190 is_shape_tensor_.resize(n_inputs, false); in SetShapeTensorMask()
192 for (int i = 0; i < n_inputs; i++) { in SetShapeTensorMask()
285 int n_inputs = GetNumberOfEngineInputs(engine); in RestoreProfiles() local
287 << n_inputs << " inputs"; in RestoreProfiles()
290 for (int j = 0; j < n_inputs; j++) { in RestoreProfiles()
|
| D | trt_engine_utils.cc | 140 int n_inputs = ctx ? ctx->num_inputs() : (input_vec ? input_vec->size() : 0); in SetTrtEngineInputs() local
142 for (int i = 0; i < n_inputs; i++) { in SetTrtEngineInputs()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.cc | 72 const int n_inputs = op_inputs.size(); in Init() local
73 if (n_inputs > 0) { in Init()
76 for (int i = 0; i < n_inputs; ++i) { in Init()
|
| D | execute.cc | 187 const int n_inputs = op->Inputs().size(); in ValidateInputTypeAndPlacement() local
188 if (kernel->num_inputs() != n_inputs) { in ValidateInputTypeAndPlacement()
190 " inputs, got ", n_inputs); in ValidateInputTypeAndPlacement()
193 if (n_inputs > 0) { in ValidateInputTypeAndPlacement()
197 for (int i = 0; i < n_inputs; ++i) { in ValidateInputTypeAndPlacement()
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | gen_node.cc | 60 int n_inputs = node_->input_size(); in ParseInputs() local
91 for (int i = 0; i < n_inputs; ++i) { in ParseInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 186 const int n_inputs = network->getNbInputs(); in GetNetworkInputShapes() local
187 input_shapes->resize(n_inputs); in GetNetworkInputShapes()
188 for (int i = 0; i < n_inputs; i++) { in GetNetworkInputShapes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 112 std::vector<NodeOut> n_inputs(num_x); in AddSymGrad() local
115 n_inputs[e->dst_input()] = {e->src(), e->src_output()}; in AddSymGrad()
118 for (const NodeOut& nout : n_inputs) { in AddSymGrad()
|
| /external/compiler-rt/lib/sanitizer_common/ |
| D | sanitizer_common_interceptors_format.inc | 301 // Stops when "consuming" n_inputs input items.
302 static void scanf_common(void *ctx, int n_inputs, bool allowGnuMalloc,
304 CHECK_GT(n_inputs, 0);
334 --n_inputs;
335 if (n_inputs < 0)
|
| /external/llvm-project/compiler-rt/lib/sanitizer_common/ |
| D | sanitizer_common_interceptors_format.inc | 300 // Stops when "consuming" n_inputs input items.
301 static void scanf_common(void *ctx, int n_inputs, bool allowGnuMalloc,
303 CHECK_GT(n_inputs, 0);
333 --n_inputs;
334 if (n_inputs < 0)
|