Searched refs:TRT_TensorOrWeights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 262 class TRT_TensorOrWeights {
264 TRT_TensorOrWeights() {} in TRT_TensorOrWeights() function
269 explicit TRT_TensorOrWeights(nvinfer1::ITensor* tensor, int batch_size = -1);
277 explicit TRT_TensorOrWeights(nvinfer1::DataType trt_dtype,
281 explicit TRT_TensorOrWeights(const TRT_ShapedWeights& weights);
283 TRT_TensorOrWeights(const TRT_TensorOrWeights& rhs);
285 void operator=(const TRT_TensorOrWeights& rhs);
361 const std::vector<TRT_TensorOrWeights>& inputs,
362 std::vector<TRT_TensorOrWeights>* outputs,
369 const std::vector<TRT_TensorOrWeights>& inputs,
[all …]
|
| D | convert_nodes.cc | 360 Status GetTrtBroadcastShape(const TRT_TensorOrWeights& operand_l, in GetTrtBroadcastShape()
361 const TRT_TensorOrWeights& operand_r, in GetTrtBroadcastShape()
396 const TRT_TensorOrWeights& input, in GetTrtBroadcastShape()
666 Status VerifyShapesMatch(absl::Span<const TRT_TensorOrWeights> inputs, in VerifyShapesMatch()
764 class TRT_TensorOrWeights::SimpleITensor : public nvinfer1::ITensor {
831 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::ITensor* tensor, in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
838 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
846 TRT_TensorOrWeights::TRT_TensorOrWeights(const TRT_ShapedWeights& weights) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
849 TRT_TensorOrWeights::TRT_TensorOrWeights(const TRT_TensorOrWeights& rhs) in TRT_TensorOrWeights() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
857 void TRT_TensorOrWeights::operator=(const TRT_TensorOrWeights& rhs) { in operator =()
[all …]
|
| D | convert_nodes_test.cc | 455 TRT_TensorOrWeights tw; in TEST()
456 TRT_TensorOrWeights copy(tw); in TEST()
457 TRT_TensorOrWeights assigned; in TEST()
472 TRT_TensorOrWeights tw(&itensor); in TEST()
473 TRT_TensorOrWeights tw1(&itensor, /*batch_size=*/1); in TEST()
476 TRT_TensorOrWeights copy(*original_ptr); in TEST()
477 TRT_TensorOrWeights assigned; in TEST()
498 TRT_TensorOrWeights tw(nvinfer1::DataType::kFLOAT, dims, /*batch_size=*/1); in TEST()
499 TRT_TensorOrWeights copy(tw); in TEST()
500 TRT_TensorOrWeights assigned; in TEST()
[all …]
|