Searched refs:CreateCudnnTensor (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 | 3486 port::StatusOr<cudnn_frontend::Tensor> CreateCudnnTensor( in CreateCudnnTensor() function
3530 CreateCudnnTensor(input_dims, input_strides, 'x', in GetCudnnOperationGraph()
3542 CreateCudnnTensor(output_dims, output_strides, 'y', in GetCudnnOperationGraph()
3554 CreateCudnnTensor(filter_dims, filter_strides, 'w', in GetCudnnOperationGraph()
3665 CreateCudnnTensor(input_dims, input_strides, 'x', in GetCudnnFusedOperationGraph()
3675 CreateCudnnTensor(output_dims, output_strides, 'y', in GetCudnnFusedOperationGraph()
3679 CreateCudnnTensor(output_dims, output_strides, 'z', in GetCudnnFusedOperationGraph()
3689 CreateCudnnTensor(filter_dims, filter_strides, 'w', in GetCudnnFusedOperationGraph()
3711 CreateCudnnTensor(bias_dims, bias_strides, 'b', bias_type, in GetCudnnFusedOperationGraph()
3718 CreateCudnnTensor(output_dims, output_strides, 'C', accumulator_type, in GetCudnnFusedOperationGraph()
[all …]
|