Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h254 class TRT_TensorOrWeights {
256 TRT_TensorOrWeights() {} in TRT_TensorOrWeights() function
261 explicit TRT_TensorOrWeights(nvinfer1::ITensor* tensor, int batch_size = -1);
269 explicit TRT_TensorOrWeights(nvinfer1::DataType trt_dtype,
273 explicit TRT_TensorOrWeights(const TRT_ShapedWeights& weights);
275 TRT_TensorOrWeights(const TRT_TensorOrWeights& rhs);
277 void operator=(const TRT_TensorOrWeights& rhs);
349 const std::vector<TRT_TensorOrWeights>& arg_inputs, in OpConverterParams()
350 std::vector<TRT_TensorOrWeights>* arg_outputs, in OpConverterParams()
361 const std::vector<TRT_TensorOrWeights>& inputs;
[all …]
Dconvert_nodes.cc363 const TRT_TensorOrWeights& operand_l, const TRT_TensorOrWeights& operand_r, in GetTrtBroadcastShape()
396 auto compute_output_dims = [](const TRT_TensorOrWeights& input, in GetTrtBroadcastShape()
643 class TRT_TensorOrWeights::SimpleITensor : public nvinfer1::ITensor {
700 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::ITensor* tensor, in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
707 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
715 TRT_TensorOrWeights::TRT_TensorOrWeights(const TRT_ShapedWeights& weights) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
718 TRT_TensorOrWeights::TRT_TensorOrWeights(const TRT_TensorOrWeights& rhs) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
726 void TRT_TensorOrWeights::operator=(const TRT_TensorOrWeights& rhs) { in operator =()
735 nvinfer1::ITensor* TRT_TensorOrWeights::tensor() { in tensor()
740 const nvinfer1::ITensor* TRT_TensorOrWeights::tensor() const { in tensor()
[all …]
Dconvert_nodes_test.cc323 TRT_TensorOrWeights tw; in TEST()
324 TRT_TensorOrWeights copy(tw); in TEST()
325 TRT_TensorOrWeights assigned; in TEST()
340 TRT_TensorOrWeights tw(&itensor); in TEST()
341 TRT_TensorOrWeights tw1(&itensor, /*batch_size=*/1); in TEST()
344 TRT_TensorOrWeights copy(*original_ptr); in TEST()
345 TRT_TensorOrWeights assigned; in TEST()
366 TRT_TensorOrWeights tw(nvinfer1::DataType::kFLOAT, dims, /*batch_size=*/1); in TEST()
367 TRT_TensorOrWeights copy(tw); in TEST()
368 TRT_TensorOrWeights assigned; in TEST()
[all …]