Searched defs:constant_tensor (Results 1 – 8 of 8) sorted by relevance
89 nvinfer1::IConstantLayer *constant_tensor = network()->addConstant(dims, weights); in ConvertTo0DTensor() local111 nvinfer1::IConstantLayer *constant_tensor = network()->addConstant(dims, weights); in ConvertTo1DTensor() local
256 nvinfer1::IConstantLayer *constant_tensor = ctx->network()->addConstant(dims, weights); in ConvertConstantTensor() local274 nvinfer1::IConstantLayer *constant_tensor = ctx->network()->addConstant(dims, weights); in ConvertScalarToITensor() local394 nvinfer1::IConstantLayer *constant_tensor = ctx->network()->addConstant(dims, weights); in ConvertTensorWithExpandDims() local410 nvinfer1::IConstantLayer *constant_tensor = ctx->network()->addConstant(dims, weights); in ConvertConstantTensor1D() local
61 void CreateSplitConstantTensors(const tensor::TensorPtr &constant_tensor, const std::vector<int64_t… in CreateSplitConstantTensors()118 …[&](const tensor::TensorPtr &constant_tensor) { return (reinterpret_cast<char *>(constant_tensor->… in CreateSplitConstantTensors()434 auto constant_tensor = GetTensorInfo(constant_node); in CreateConstantOutputsOfSplit() local
134 nvinfer1::IConstantLayer *constant_tensor; in AddInnerOp() local
88 nvinfer1::IConstantLayer *constant_tensor = ctx->network()->addConstant(dims, weights); in CastTensor() local
116 nvinfer1::IConstantLayer *constant_tensor = ctx->network()->addConstant(dims, weights); in CastTensor() local
1083 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1140 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1246 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1304 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1340 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local
639 auto constant_tensor = constant_value_node->value()->cast<tensor::TensorPtr>(); in operator() local