/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
D | trt_plugin.cc | 40 std::memcpy(&(dim.nbDims), buffer, sizeof(dim.nbDims)); in PluginTensorRT() 41 buffer += sizeof(dim.nbDims); in PluginTensorRT() 55 dim.nbDims = inputs[index].nbDims; in configure() 56 for (int i = 0; i < dim.nbDims; i++) { in configure() 67 sizeof(input_dim_list_.size()) + sizeof(dim.nbDims) + sizeof(dim.d) + in getSerializationSize() 86 std::memcpy(buffer, &(dim.nbDims), sizeof(dim.nbDims)); in serialize() 87 buffer += sizeof(dim.nbDims); in serialize()
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.cc | 209 trt_dims.nbDims = shape.dims() - offset; in TensorShapeToTrtDims() 224 if (dims->nbDims < 2) { in RemoveBatchDimension() 228 std::copy(dims->d + 1, dims->d + dims->nbDims, dims->d); in RemoveBatchDimension() 229 dims->nbDims--; in RemoveBatchDimension() 338 string out = StrCat("nvinfer1::Dims(nbDims=", dims.nbDims, ", d="); in DebugString() 339 for (int i = 0; i < dims.nbDims; ++i) { in DebugString() 401 std::copy(input_dims.d, input_dims.d + input_dims.nbDims, in GetTrtBroadcastShape() 402 output_dims_array + broadcast_num_dims - input_dims.nbDims); in GetTrtBroadcastShape() 404 const int true_input_dims = input_dims.nbDims + 1; in GetTrtBroadcastShape() 416 output_dims->nbDims = broadcast_num_dims - 1; in GetTrtBroadcastShape() [all …]
|
D | convert_nodes_test.cc | 79 dims.nbDims = d.size(); in GetTestDims() 145 if (lhs.nbDims != rhs.nbDims) return false; in TrtDimsEquals() 146 for (int i = 0; i < lhs.nbDims; ++i) { in TrtDimsEquals() 337 dims.nbDims = 1; in TEST() 364 dims.nbDims = 1; in TEST() 1186 if (dims.nbDims < 0) return false; in HasStaticShape() 1187 for (int i = 0; i < dims.nbDims; ++i) { in HasStaticShape()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cudnn_6_0.inc | 120 int nbDims, 126 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA); 133 int nbDims, 138 return func_ptr(tensorDesc, format, dataType, nbDims, dimA); 145 int *nbDims, 151 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA); 402 int nbDims, 407 return func_ptr(filterDesc, dataType, format, nbDims, filterDimA); 415 int *nbDims, 420 return func_ptr(filterDesc, nbDimsRequested, dataType, format, nbDims, filterDimA); [all …]
|
D | cudnn_7_0.inc | 131 int nbDims, 137 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA); 144 int nbDims, 149 return func_ptr(tensorDesc, format, dataType, nbDims, dimA); 156 int *nbDims, 162 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA); 413 int nbDims, 418 return func_ptr(filterDesc, dataType, format, nbDims, filterDimA); 426 int *nbDims, 431 return func_ptr(filterDesc, nbDimsRequested, dataType, format, nbDims, filterDimA); [all …]
|
D | cudnn_7_3.inc | 136 int nbDims, 142 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA); 149 int nbDims, 154 return func_ptr(tensorDesc, format, dataType, nbDims, dimA); 161 int *nbDims, 167 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA); 411 int nbDims, 416 return func_ptr(filterDesc, dataType, format, nbDims, filterDimA); 424 int *nbDims, 429 return func_ptr(filterDesc, nbDimsRequested, dataType, format, nbDims, filterDimA); [all …]
|
D | cudnn_7_1.inc | 131 int nbDims, 137 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA); 144 int nbDims, 149 return func_ptr(tensorDesc, format, dataType, nbDims, dimA); 156 int *nbDims, 162 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA); 413 int nbDims, 418 return func_ptr(filterDesc, dataType, format, nbDims, filterDimA); 426 int *nbDims, 431 return func_ptr(filterDesc, nbDimsRequested, dataType, format, nbDims, filterDimA); [all …]
|
D | cudnn_7_4.inc | 136 int nbDims, 142 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA); 149 int nbDims, 154 return func_ptr(tensorDesc, format, dataType, nbDims, dimA); 161 int *nbDims, 167 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA); 411 int nbDims, 416 return func_ptr(filterDesc, dataType, format, nbDims, filterDimA); 424 int *nbDims, 429 return func_ptr(filterDesc, nbDimsRequested, dataType, format, nbDims, filterDimA); [all …]
|
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
D | inc_op_plugin.cc | 72 for (int i = 0; i < input_dim_list_[0].nbDims; i++) { in enqueue()
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
D | trt_engine_op.cc | 453 std::vector<int> trt_shape(dims.nbDims + 1); in ExecuteTrtEngine() 455 for (int j = 0; j < dims.nbDims; j++) trt_shape[j + 1] = dims.d[j]; in ExecuteTrtEngine()
|