Searched refs:GetTensorVectorSizeAndDim (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 3454 std::tuple<int, int> GetTensorVectorSizeAndDim( in GetTensorVectorSizeAndDim() function
3470 std::tuple<int, int> GetTensorVectorSizeAndDim( in GetTensorVectorSizeAndDim() function
3518 GetTensorVectorSizeAndDim(input_descriptor, input_type); in GetCudnnOperationGraph()
3535 GetTensorVectorSizeAndDim(output_descriptor, output_type); in GetCudnnOperationGraph()
3547 GetTensorVectorSizeAndDim(filter_descriptor, input_type); in GetCudnnOperationGraph()
3653 GetTensorVectorSizeAndDim(input_descriptor, input_type); in GetCudnnFusedOperationGraph()
3669 GetTensorVectorSizeAndDim(output_descriptor, output_type); in GetCudnnFusedOperationGraph()
3683 GetTensorVectorSizeAndDim(filter_descriptor, input_type); in GetCudnnFusedOperationGraph()
3705 GetTensorVectorSizeAndDim(bias_descriptor, input_type); in GetCudnnFusedOperationGraph()
3715 GetTensorVectorSizeAndDim(output_descriptor, output_type); in GetCudnnFusedOperationGraph()
|