Searched refs:use_implicit_batch_ (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 225 bool use_implicit_batch_; member in tensorflow::tensorrt::TRTEngineOp
475 status = context->GetAttr("_use_implicit_batch", &use_implicit_batch_); in TRTEngineOp()
479 use_implicit_batch_ = true; in TRTEngineOp()
493 if (use_implicit_batch_) { in TRTEngineOp()
711 if (!use_implicit_batch_) { in VerifyInputShapes()
749 if (use_implicit_batch_) { in VerifyInputShapes()
829 if (!use_implicit_batch_ && in ComputeAsync()
882 if (!use_implicit_batch_ && (has_dynamic_shape_input_ || in ComputeAsync()
984 use_implicit_batch_ ? ctx->input(0).shape().dim_size(0) : 0; in ExecuteTrtEngine()
988 use_implicit_batch_, num_batch, profiles, ctx)); in ExecuteTrtEngine()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.h | 241 const bool use_implicit_batch_; variable
317 bool use_implicit_batch() const { return use_implicit_batch_; } in use_implicit_batch()
482 const bool use_implicit_batch_; variable
|
| H A D | convert_nodes.cc | 875 use_implicit_batch_(use_implicit_batch), in TrtNodeValidator()
938 node_def.op(), dtype, shape, use_implicit_batch_, in ConvertToTensorOrWeights()
997 use_implicit_batch_, use_explicit_precision_); in IsTensorRTCandidate()
1008 use_implicit_batch_, use_explicit_precision_); in ConvertConstToWeights()
1039 use_implicit_batch_(use_implicit_batch), in Converter()
1051 use_implicit_batch_ in Init()
1121 if (use_implicit_batch_) { in AddInputTensor()
1289 if (!use_implicit_batch_ && profiles) { in BuildCudaEngine()
1395 if (use_implicit_batch_ && input.is_tensor()) { in AddTensorOrWeights()
1418 const int order_size = use_implicit_batch_ ? order_with_batch_dim.size() - 1 in TransposeTensor()
[all …]
|