Searched refs:use_explicit_precision_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 243 const bool use_explicit_precision_; variable
428 bool UseExplicitPrecision() const { return use_explicit_precision_; } in UseExplicitPrecision()
498 bool use_explicit_precision_; variable
|
| D | convert_nodes.cc | 876 use_explicit_precision_(use_explicit_precision) {} in TrtNodeValidator()
997 use_implicit_batch_, use_explicit_precision_); in IsTensorRTCandidate()
1008 use_implicit_batch_, use_explicit_precision_); in ConvertConstToWeights()
1041 use_explicit_precision_(use_explicit_precision) { in Converter()
1055 if (use_explicit_precision_) { in Init()
1280 if (IS_TRT_VERSION_GE(8, 0, 0, 0) || !use_explicit_precision_) { in BuildCudaEngine()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 275 bool use_explicit_precision_; member in tensorflow::tensorrt::TRTEngineOp
452 status = context->GetAttr("use_explicit_precision", &use_explicit_precision_); in TRTEngineOp()
454 use_explicit_precision_ = false; in TRTEngineOp()
1074 &cache_resource->profiles_, name(), use_explicit_precision_, &cluster); in BuildEngine()
1360 /*use_explicit_precision=*/use_explicit_precision_, in AllocateCalibrationResources()
|