Searched refs:use_implicit_batch_ (Results 1 – 5 of 5) sorted by relevance
175 bool use_implicit_batch_; member in tensorflow::tensorrt::TRTEngineOp390 status = context->GetAttr("_use_implicit_batch", &use_implicit_batch_); in TRTEngineOp()394 use_implicit_batch_ = true; in TRTEngineOp()397 if (!use_implicit_batch_) { in TRTEngineOp()400 use_implicit_batch_ = true; in TRTEngineOp()411 if (use_implicit_batch_) { in TRTEngineOp()540 if (!use_implicit_batch_) { in VerifyInputShapes()577 if (use_implicit_batch_) { in VerifyInputShapes()674 if (!use_implicit_batch_ && has_dynamic_shape_input_) { in ComputeAsync()782 use_implicit_batch_ ? ctx->input(0).shape().dim_size(0) : 0; in ExecuteTrtEngine()[all …]
44 use_implicit_batch_(true), in name_()77 bool use_implicit_batch_; variable
434 const bool use_implicit_batch_; variable500 bool use_implicit_batch() const { return use_implicit_batch_; } in use_implicit_batch()671 const bool use_implicit_batch_; variable
77 use_implicit_batch_ = params.at("use_implicit_batch").b(); in Init()244 cp.use_implicit_batch = use_implicit_batch_; in Optimize()
1143 use_implicit_batch_(use_implicit_batch) { in TrtNodeValidator()1178 node_def.op(), dtype, shape, use_implicit_batch_, in ConvertToTensorOrWeights()1224 use_implicit_batch_); in IsTensorRTCandidate()1235 use_implicit_batch_); in ConvertConstToWeights()1259 use_implicit_batch_(use_implicit_batch), in Converter()1272 use_implicit_batch_ in Init()1338 if (use_implicit_batch_) { in AddInputTensor()1496 if (!use_implicit_batch_ && profiles) { in BuildCudaEngine()1563 if (use_implicit_batch_ && input.is_tensor()) { in AddTensorOrWeights()1586 const int order_size = use_implicit_batch_ ? order_with_batch_dim.size() - 1 in TransposeTensor()[all …]