Searched refs:dimvec (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles_test.cc | 34 std::vector<TensorShape> DimVecToShapeVec(std::vector<nvinfer1::Dims3> dimvec) { in DimVecToShapeVec() argument
35 std::vector<TensorShape> shapevec(dimvec.size()); in DimVecToShapeVec()
36 for (int i = 0; i < dimvec.size(); i++) { in DimVecToShapeVec()
39 TensorShapeUtils::MakeShape(dimvec[i].d, dimvec[i].nbDims, &shape)); in DimVecToShapeVec()
191 for (auto dimvec : input_profiles) { in TEST_F() local
192 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dimvec); in TEST_F()
198 for (int j = 0; j < dimvec.size(); j++) { in TEST_F()
207 EXPECT_TRUE(DimsContained(dimvec[j], min, max)); in TEST_F()
212 EXPECT_TRUE(DimsEqual(dimvec[j], opt)); in TEST_F()
|
| D | trt_shape_optimization_profiles.cc | 31 std::vector<nvinfer1::Dims> dimvec(shape_vec.size()); in GetDimVec() local
32 absl::c_transform(shape_vec, dimvec.begin(), [](TensorShapeType shape) { in GetDimVec()
35 return dimvec; in GetDimVec()
51 const std::vector<nvinfer1::Dims>& dimvec, std::vector<nvinfer1::Dims>* min, in SetImplicitBatchModeCompatibleProfile() argument
53 *min = dimvec; in SetImplicitBatchModeCompatibleProfile()
57 *opt = dimvec; in SetImplicitBatchModeCompatibleProfile()
58 *max = dimvec; in SetImplicitBatchModeCompatibleProfile()
64 std::vector<nvinfer1::Dims> dimvec = GetDimVec(shape_vec); in ImplicitBatchModeCompatibleStrategy() local
66 SetImplicitBatchModeCompatibleProfile(dimvec, &min, &opt, &max); in ImplicitBatchModeCompatibleStrategy()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 72 std::vector<tensorflow::int64> dimvec(num_dims); in CreateTensor() local
74 dimvec[i] = static_cast<tensorflow::int64>(dims[i]); in CreateTensor()
80 tensorflow::TensorShape(dimvec), buf)); in CreateTensor()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 139 string DebugString(const std::vector<nvinfer1::Dims>& dimvec) { in DebugString() argument
141 absl::StrJoin(dimvec, ",", in DebugString()
|
| D | utils.h | 110 string DebugString(const std::vector<nvinfer1::Dims>& dimvec);
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental.cc | 579 std::vector<tensorflow::int64> dimvec(num_dims); in TFE_AllocateHostTensor() local
581 dimvec[i] = static_cast<tensorflow::int64>(dims[i]); in TFE_AllocateHostTensor()
591 static_cast<tensorflow::DataType>(dtype), dimvec); in TFE_AllocateHostTensor()
|
| D | c_api.cc | 578 std::vector<tensorflow::int64> dimvec(num_dims); in TFE_NewTensorHandleFromDeviceMemory() local
580 dimvec[i] = static_cast<tensorflow::int64>(dims[i]); in TFE_NewTensorHandleFromDeviceMemory()
589 [dimvec](int dim_index, TF_Status* status) { in TFE_NewTensorHandleFromDeviceMemory()
590 return dimvec[dim_index]; in TFE_NewTensorHandleFromDeviceMemory()
605 tensorflow::TensorShape(dimvec), buf); in TFE_NewTensorHandleFromDeviceMemory()
|