Searched defs:batch_dim_ (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ | ||
| H A D | reverse_sequence_op.h | 52 int32 batch_dim_; variable |
| H A D | reverse_sequence_op.cc | 166 int32 batch_dim_; member in tensorflow::ReverseSequenceOp |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
| H A D | reverse_sequence_op.cc | 127 int32 batch_dim_; member in tensorflow::__anon254a0c5e0111::ReverseSequenceOp |