Home
last modified time | relevance | path

Searched refs:nbDims (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dtrt_plugin.cc40 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/
Dconvert_nodes.cc209 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 …]
Dconvert_nodes_test.cc79 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/
Dcudnn_6_0.inc120 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 …]
Dcudnn_7_0.inc131 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 …]
Dcudnn_7_3.inc136 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 …]
Dcudnn_7_1.inc131 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 …]
Dcudnn_7_4.inc136 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/
Dinc_op_plugin.cc72 for (int i = 0; i < input_dim_list_[0].nbDims; i++) { in enqueue()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc453 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()