Home
last modified time | relevance | path

Searched refs:TRT_TensorOrWeights (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dweights.cc84 TRT_TensorOrWeights::TRT_TensorOrWeights(ITensorProxyPtr tensor) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
89 TRT_TensorOrWeights::TRT_TensorOrWeights(ITensorProxyPtr tensor, int batch_size) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
95 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::ITensor* tensor, in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
102 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
110 TRT_TensorOrWeights::TRT_TensorOrWeights(const TRT_ShapedWeights& weights) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
115 TRT_TensorOrWeights::TRT_TensorOrWeights(const ResourceHandle& resource) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
120 TRT_TensorOrWeights::TRT_TensorOrWeights(const TRT_TensorOrWeights& rhs) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
128 void TRT_TensorOrWeights::operator=(const TRT_TensorOrWeights& rhs) { in operator =()
137 ITensorProxyPtr TRT_TensorOrWeights::tensor() const { in tensor()
142 ResourceHandle TRT_TensorOrWeights::resource() const { in resource()
[all …]
Dweights.h180 class TRT_TensorOrWeights {
182 TRT_TensorOrWeights() {} in TRT_TensorOrWeights() function
183 TRT_TensorOrWeights(ITensorProxyPtr);
184 TRT_TensorOrWeights(ITensorProxyPtr tensor, int batch_size);
190 explicit TRT_TensorOrWeights(nvinfer1::ITensor* tensor, int batch_size = -1);
198 explicit TRT_TensorOrWeights(nvinfer1::DataType trt_dtype,
202 explicit TRT_TensorOrWeights(const TRT_ShapedWeights& weights);
205 explicit TRT_TensorOrWeights(const ResourceHandle& resource);
207 TRT_TensorOrWeights(const TRT_TensorOrWeights& rhs);
209 void operator=(const TRT_TensorOrWeights& rhs);
Dconvert_nodes.h213 const std::vector<TRT_TensorOrWeights>& inputs,
214 TRT_TensorOrWeights* output);
219 const std::vector<TRT_TensorOrWeights>& inputs,
220 TRT_TensorOrWeights* output);
226 TRT_TensorOrWeights* tensor_or_weights);
424 std::unordered_map<string, TRT_TensorOrWeights>& TensorsMap() { in TensorsMap()
443 Status AddTensorOrWeights(const string& name, TRT_TensorOrWeights input);
446 Status GetTensorOrWeights(const string& name, TRT_TensorOrWeights* output);
450 std::vector<TRT_TensorOrWeights>* inputs) const;
453 std::unordered_map<string, TRT_TensorOrWeights> trt_tensors_;
[all …]
Dop_converter.h44 const std::vector<TRT_TensorOrWeights>& inputs,
45 std::vector<TRT_TensorOrWeights>* outputs,
52 const std::vector<TRT_TensorOrWeights>& inputs,
53 std::vector<TRT_TensorOrWeights>* outputs,
58 const std::vector<TRT_TensorOrWeights>& inputs;
59 std::vector<TRT_TensorOrWeights>* outputs;
175 void AddOutput(const TRT_TensorOrWeights& out) { in AddOutput()
Dconvert_nodes.cc276 Status GetTrtBroadcastShape(const TRT_TensorOrWeights& operand_l, in GetTrtBroadcastShape()
277 const TRT_TensorOrWeights& operand_r, in GetTrtBroadcastShape()
314 [use_implicit_batch](const TRT_TensorOrWeights& input, in GetTrtBroadcastShape()
406 Status BroadcastWeights(std::unique_ptr<TRT_TensorOrWeights>& p, in BroadcastWeights()
412 p = std::make_unique<TRT_TensorOrWeights>(weights); in BroadcastWeights()
417 Status ApplyBroadcast(std::unique_ptr<TRT_TensorOrWeights>& operand, in ApplyBroadcast()
440 operand = std::make_unique<TRT_TensorOrWeights>(tensor); in ApplyBroadcast()
452 Status BroadcastTensors(std::unique_ptr<TRT_TensorOrWeights>& operand_l, in BroadcastTensors()
453 std::unique_ptr<TRT_TensorOrWeights>& operand_r, in BroadcastTensors()
633 Status VerifyShapesMatch(absl::Span<const TRT_TensorOrWeights> inputs, in VerifyShapesMatch()
[all …]
Dop_converter_test.cc56 AddOutput(TRT_TensorOrWeights(nvinfer1::DataType::kFLOAT, in Convert()
111 .insert(std::make_pair("input2", TRT_TensorOrWeights(TRT_ShapedWeights( in TEST()
Dconvert_nodes_test.cc235 TRT_TensorOrWeights tw; in TEST()
236 TRT_TensorOrWeights copy(tw); in TEST()
237 TRT_TensorOrWeights assigned; in TEST()
252 TRT_TensorOrWeights tw(itensor); in TEST()
253 TRT_TensorOrWeights tw1(itensor, /*batch_size=*/1); in TEST()
256 TRT_TensorOrWeights copy(*original_ptr); in TEST()
257 TRT_TensorOrWeights assigned; in TEST()
278 TRT_TensorOrWeights tw(nvinfer1::DataType::kFLOAT, dims, /*batch_size=*/1); in TEST()
279 TRT_TensorOrWeights copy(tw); in TEST()
280 TRT_TensorOrWeights assigned; in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dfill_ops.cc65 const TRT_TensorOrWeights& dims_input = inputs.at(0); in Validate()
93 const TRT_TensorOrWeights& dims_input = inputs.at(0); in Convert()
94 const TRT_TensorOrWeights& value_input = inputs.at(1); in Convert()
110 this->AddOutput(TRT_TensorOrWeights(output_tensor)); in Convert()
212 const TRT_TensorOrWeights& input = inputs.at(0); in Convert()
213 TRT_TensorOrWeights value_input; in Convert()
251 value_input = TRT_TensorOrWeights(value_weights.ValueOrDie()); in Convert()
260 TRT_TensorOrWeights dims_input(dims_input_tensor); in Convert()
271 this->AddOutput(TRT_TensorOrWeights(output_tensor)); in Convert()
277 float get_input_param(const TRT_TensorOrWeights& input) { in get_input_param()
Dlike_ops.cc65 const TRT_TensorOrWeights &input = inputs.at(0); in Convert()
74 TRT_TensorOrWeights value_input(value_weights.ValueOrDie()); in Convert()
87 TRT_TensorOrWeights dims_input(dims_input_tensor); in Convert()
92 this->AddOutput(TRT_TensorOrWeights(output_tensor)); in Convert()
Deinsum.cc190 Status Initialize(const TRT_TensorOrWeights& operand, Labels input_labels, in Initialize()
255 const TRT_TensorOrWeights& operand, Labels input_labels, in Create()
295 const TRT_TensorOrWeights& operand) { in SetDynamicSize()
406 StatusOr<TRT_TensorOrWeights> ConditionEinsumWeights( in ConditionEinsumWeights()
407 TRTNetworkBuilder* builder, const TRT_TensorOrWeights& operand, in ConditionEinsumWeights()
417 return TRT_TensorOrWeights(weights); in ConditionEinsumWeights()
424 return TRT_TensorOrWeights((*tensor)->getOutput(0)); in ConditionEinsumWeights()
430 std::unique_ptr<TRT_TensorOrWeights>* operand, in ConditionEinsumTensor()
457 *operand = std::make_unique<TRT_TensorOrWeights>(*shuffle_out); in ConditionEinsumTensor()
465 std::unique_ptr<TRT_TensorOrWeights>* operand, in ConditionEinsumOperand()
[all …]
Dvariable_ops.cc66 const TRT_TensorOrWeights& handle = inputs.at(0); in ReadVariableHelper()
192 AddOutput(TRT_TensorOrWeights(weights)); in ValidateImpl()
221 AddOutput(TRT_TensorOrWeights(weights)); in ConvertImpl()
290 AddOutput(TRT_TensorOrWeights(weights)); in ValidateImpl()
324 AddOutput(TRT_TensorOrWeights(weights)); in ConvertImpl()
Dquantization_ops.cc86 const TRT_TensorOrWeights& input) { in ExlicitQDQInputToTensor()
134 const std::vector<TRT_TensorOrWeights>& inputs, const NodeDef& node_def, in ValidateQDQForExplicitPrecision()
244 const std::vector<TRT_TensorOrWeights>& inputs, const NodeDef& node_def, in ValidateQDQForExplicitPrecision()
274 const std::vector<TRT_TensorOrWeights>& inputs, const NodeDef& node_def, in ValidateQDQForExplicitPrecision()
301 const std::vector<TRT_TensorOrWeights>& inputs, const NodeDef& node_def, in ValidateQDQForExplicitPrecision()
Dslice_ops.cc52 OpConverterParams* params, const TRT_TensorOrWeights& input, in ConvertStridedSliceHelper()
136 params->converter, TRT_TensorOrWeights(tensor), *final_shape, in ConvertStridedSliceHelper()
139 params->outputs->push_back(TRT_TensorOrWeights(tensor)); in ConvertStridedSliceHelper()
Ddata_format_vec_permute.cc170 params_->outputs->push_back(TRT_TensorOrWeights(output_tensor)); in Convert()
175 TRT_TensorOrWeights x_input_;
Dslice_ops.h58 OpConverterParams* params, const TRT_TensorOrWeights& input,
Dtile.cc200 params.converter, TRT_TensorOrWeights(output_tensor), in Convert()
204 AddOutput(TRT_TensorOrWeights(output_tensor)); in Convert()
Dbinary_ops.cc135 TRT_TensorOrWeights(unary_layer->getOutput(0)));
140 params.outputs->push_back(TRT_TensorOrWeights(output));
Dunary_ops.cc123 params.outputs->push_back(TRT_TensorOrWeights(layer->getOutput(0))); in ConvertImpl()
231 params_->outputs->push_back(TRT_TensorOrWeights(output_tensor)); in Convert()
Dlog_softmax.cc97 params.outputs->push_back(TRT_TensorOrWeights(sub->getOutput(0))); in Convert()
Dlayer_utils.h344 Status get_tensor4TensorOrWeights(const TRT_TensorOrWeights& input, in get_tensor4TensorOrWeights()
494 StatusOr<nvinfer1::ILayer*> AddFill(const TRT_TensorOrWeights& value_input,
495 const TRT_TensorOrWeights& dims_input,