Home
last modified time | relevance | path

Searched defs:use_padded_io (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc1102 bool use_padded_io) { in CreateRnnDescriptor()
1128 bool use_padded_io) { in GetCachedRnnDescriptor()
1506 bool use_padded_io = false; in ComputeAndReturnAlgorithm() local
1908 bool use_padded_io = false; in ComputeImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.cc381 bool use_padded_io) { in createRnnDescriptor()
H A Ddnn.h2045 ScratchAllocator* state_allocator, bool use_padded_io) { in createRnnDescriptor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1269 uint64_t seed, ScratchAllocator* state_allocator, bool use_padded_io) { in Create()
2419 bool use_padded_io) { in createRnnDescriptor()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2575 bool use_padded_io) { in createRnnDescriptor()