Home
last modified time | relevance | path

Searched refs:GetTempWeights (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dweights.h152 StatusOr<TRT_ShapedWeights> GetTempWeights(nvinfer1::DataType trt_type,
157 StatusOr<TRT_ShapedWeights> GetTempWeights(const TRT_ShapedWeights& weights) { in GetTempWeights() function
158 return GetTempWeights(weights.TrtDType(), weights.Shape()); in GetTempWeights()
Dweights.cc189 StatusOr<TRT_ShapedWeights> TrtWeightStore::GetTempWeights( in GetTempWeights() function in tensorflow::tensorrt::convert::TrtWeightStore
Dconvert_nodes.cc496 params->weight_store->GetTempWeights(trt_type, dims);
1923 StatusOr<TRT_ShapedWeights> tmp = params->weight_store->GetTempWeights( in ConvertConv2DHelper()
1968 params->weight_store->GetTempWeights(weights_rsck); in ConvertConv2DHelper()
1970 StatusOr<TRT_ShapedWeights> biases = params->weight_store->GetTempWeights( in ConvertConv2DHelper()
2864 params->weight_store->GetTempWeights(weights_drsck); in ConvertConv3DHelper()
2920 params->weight_store->GetTempWeights(weights_drsck); in ConvertConv3DHelper()
3183 params->weight_store->GetTempWeights(weights); in ConvertFusedConv2DBiasActivation()
3458 auto tmp = weight_store->GetTempWeights(trt_dtype, weight_dims->AsTrtDims()); in TfTensorToTrtWeights()
4233 params->weight_store->GetTempWeights(*ptr_shape_weights); in ConvertFusedBatchNorm()
4236 params->weight_store->GetTempWeights(*ptr_shape_weights); in ConvertFusedBatchNorm()
[all …]
Dconvert_nodes_test.cc214 store.GetTempWeights(nvinfer1::DataType::kFLOAT, CreateDims({2, 5})) in TEST()
741 ->GetTempWeights(nvinfer1::DataType::kFLOAT, CreateDims(input_dims)) in TestPrepareTensorForShape()
855 weight_store->GetTempWeights(trt_type, CreateDims({2, 3})).ValueOrDie(); in TestGetWeightRange()
1016 weight_store_->GetTempWeights(dtype, CreateDims({2, 3, 5})) in TEST_F()
1428 converter_->weight_store_.GetTempWeights(dtype, dims_adap.AsTrtDims()) in AddTestWeights()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dlike_ops.cc71 params.weight_store->GetTempWeights(input.TrtDType(), value_input_dims); in Convert()
Dlayer_utils.h255 weight_store_->GetTempWeights(nvinfer1::DataType::kINT32, shape_dims); in ConstantShape()
277 weight_store_->GetTempWeights(nvinfer1::DataType::kINT32, shape_dims); in Constant()
302 weight_store_->GetTempWeights(data_type, shape); in Constant()
370 weight_store_->GetTempWeights(data_type, weights_shape); in ScalarWeights()
Ddata_format_vec_permute.cc149 params_->weight_store->GetTempWeights(nvinfer1::DataType::kINT32, in Convert()
Dfill_ops.cc246 params.weight_store->GetTempWeights(input.TrtDType(), in Convert()