Searched defs:constant_tensor (Results 1 – 4 of 4) sorted by relevance
53 void CreateSplitConstantTensors(const tensor::TensorPtr &constant_tensor, const std::vector<int64_t… in CreateSplitConstantTensors()110 …[&](const tensor::TensorPtr &constant_tensor) { return (reinterpret_cast<char *>(constant_tensor->… in CreateSplitConstantTensors()423 auto constant_tensor = GetTensorInfo(constant_node); in CreateConstantOutputsOfSplit() local
124 nvinfer1::IConstantLayer *constant_tensor = network->addConstant(dims, weights); in ConvertConstantTensor() local138 nvinfer1::IConstantLayer *constant_tensor = network->addConstant(dims, weights); in ConvertScalarToITensor() local193 nvinfer1::IConstantLayer *constant_tensor = network->addConstant(dims, weights); in ConvertTensorWithExpandDims() local
1015 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1069 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1172 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1227 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local1260 std::shared_ptr<Tensor> constant_tensor; in TEST_F() local
668 auto constant_tensor = constant_value_node->value()->cast<tensor::TensorPtr>(); in operator() local