| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.cc | 420 const nvinfer1::ICudaEngine* engine, int n_inputs) { in SetShapeTensorMask() argument
421 is_shape_tensor_.resize(n_inputs, false); in SetShapeTensorMask()
422 for (int i = 0; i < n_inputs; i++) { in SetShapeTensorMask()
440 int n_inputs = network->getNbInputs(); in SetShapeTensorMask() local
441 is_shape_tensor_.resize(n_inputs, false); in SetShapeTensorMask()
442 for (int i = 0; i < n_inputs; i++) { in SetShapeTensorMask()
598 int n_inputs = GetNumberOfEngineInputs(engine); in RestoreProfiles() local
599 if (n_inputs > n_network_inputs) { in RestoreProfiles()
603 << n_inputs << " inputs"; in RestoreProfiles()
630 cfg.min[j + n_inputs] = GetDimsFromShapeVal( in RestoreProfiles()
[all …]
|
| D | trt_engine_utils.cc | 124 int n_inputs = ctx ? ctx->num_inputs() : (input_vec ? input_vec->size() : 0); in SetTrtEngineInputs() local
126 for (int i = 0; i < n_inputs; i++) { in SetTrtEngineInputs()
|
| D | trt_shape_optimization_profiles.h | 328 void SetShapeTensorMask(const nvinfer1::ICudaEngine* engine, int n_inputs);
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 62 int n_inputs; member
149 this->param_.n_inputs == 1 in CreateTRTEngine()
164 const int n_input = param_.n_inputs; in CreateTRTEngine()
181 if (this->param_.n_inputs == 2) { in CreateTRTEngine()
287 if (param_.n_inputs > 1) { in TEST_P()
334 EXPECT_EQ(param_.n_inputs, engine_instance.input_shapes_size()); in TEST_P()
362 if (this->param_.n_inputs == 1) { in TEST_P()
|
| /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 | 213 const int n_inputs = op->Inputs().size(); in ValidateInputTypeAndPlacement() local
214 if (kernel->num_inputs() != n_inputs) { in ValidateInputTypeAndPlacement()
216 " inputs, got ", n_inputs); in ValidateInputTypeAndPlacement()
219 if (n_inputs > 0) { in ValidateInputTypeAndPlacement()
223 for (int i = 0; i < n_inputs; ++i) { in ValidateInputTypeAndPlacement()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 143 const int n_inputs = network->getNbInputs(); in GetNetworkInputShapes() local
144 input_shapes->resize(n_inputs); in GetNetworkInputShapes()
145 for (int i = 0; i < n_inputs; i++) { in GetNetworkInputShapes()
|
| D | convert_nodes_test.cc | 3383 int n_inputs = 2) -> NodeDef { in TEST_P() argument
3387 if (n_inputs > 1) { in TEST_P()
3601 int n_inputs = p.shape_b.empty() ? 1 : 2; in TEST_P() local
3602 NodeDef node_def = get_einsum_nodedef(tf_type_, p.equation, n_inputs); in TEST_P()
|
| D | convert_nodes.cc | 1335 int n_inputs = network()->getNbInputs(); in BuildCudaEngine() local
1336 for (int i = 0; i < n_inputs; i++) { in BuildCudaEngine()
|
| /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/ |
| D | trt_convert_api.cc | 341 Status ValidateConversionParams(const TfTrtConversionParams& p, int n_inputs) { in ValidateConversionParams() argument
347 if (p.convert_to_static_engine && n_inputs == 0) { in ValidateConversionParams()
352 if (!p.convert_to_static_engine && n_inputs >= 0) { in ValidateConversionParams()
364 if (!p.allow_build_at_runtime && n_inputs == 0) { in ValidateConversionParams()
|
| /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/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 190 const size_t n_inputs = input_tf_tensors.size(); in ValidateInputTypes() local
192 if (input_types.size() != n_inputs) { in ValidateInputTypes()
194 " inputs, got ", n_inputs); in ValidateInputTypes()
197 for (size_t i = 0; i < n_inputs; ++i) { in ValidateInputTypes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3679 int n_inputs = inputs_ty.size(); in verify() local
3680 if (n_inputs < 1) return op.emitOpError() << "No inputs"; in verify()
3684 if (n_init_values != n_inputs) { in verify()
3685 return op.emitOpError() << "Number of inputs (" << n_inputs in verify()
3692 for (int i = 1; i < n_inputs; ++i) { in verify()
3709 for (int i = 0; i < n_inputs; ++i) { in verify()
3735 int n_inputs = inputs_ty.size(); in verify() local
3738 for (int i = 1; i < n_inputs; ++i) { in verify()
|