Searched refs:params_size_in_bytes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 1043 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local
1044 CHECK(params_size_in_bytes % sizeof(T) == 0) in Compute()
1046 int64 params_size = params_size_in_bytes / sizeof(T); in Compute()
1087 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local
1088 CHECK(params_size_in_bytes % sizeof(T) == 0) in Compute()
1215 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local
1216 CHECK(params_size_in_bytes % sizeof(T) == 0) in Compute()
1219 int params_size = params_size_in_bytes / sizeof(T); in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 968 CudnnRnnParamsDescriptor(FilterDescriptor handle, int64 params_size_in_bytes, in CudnnRnnParamsDescriptor() argument
971 params_size_in_bytes_(params_size_in_bytes), in CudnnRnnParamsDescriptor()
984 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::__anonf41f84630311::CudnnRnnParamsDescriptor
1121 return params_desc_.params_size_in_bytes(); in ParamsSizeInBytes()
1175 int64 params_size_in_bytes = static_cast<int64>(params_size); in Create() local
1178 int filter_dims[] = {static_cast<int>(params_size_in_bytes), 1, 1}; in Create()
1245 return CudnnRnnParamsDescriptor(std::move(filter_desc), params_size_in_bytes, in Create()
1472 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local
1475 /*xDesc=*/input_desc.handles()[0], /*sizeInBytes=*/¶ms_size_in_bytes, in CheckRNNParameterSize()
1477 if (static_cast<int64>(params_size_in_bytes) != in CheckRNNParameterSize()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1658 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::MIOpenRnnParamsDescriptor
1732 return miopen_params_desc_->params_size_in_bytes(); in ParamsSizeInBytes()
1954 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local
1957 input_desc.handles()[0] /*xDesc*/, ¶ms_size_in_bytes /*sizeInBytes*/, in CheckRNNParameterSize()
1963 return static_cast<int64>(params_size_in_bytes) == in CheckRNNParameterSize()
|