| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cholesky_thunk.h | 66 const int64 batch_size_; variable
|
| D | triangular_solve_thunk.h | 65 const int64 batch_size_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | record_input_op.cc | 66 int64 batch_size_; member in tensorflow::RecordInputOp
|
| D | word2vec_kernels.cc | 112 int32 batch_size_ = 0; member in tensorflow::SkipgramOp
|
| D | cudnn_rnn_ops.cc | 211 const int batch_size_; member in tensorflow::__anon454a70830111::CudnnRnnParameters
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_decoder.h | 66 int batch_size_; variable
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_decoder.h | 66 int batch_size_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 335 int batch_size_ = -1; variable
579 int batch_size_ = -1; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 248 const int64 batch_size_; member in tensorflow::data::__anonb889a47b0111::BatchDatasetOp::Dataset
|
| D | padded_batch_dataset_op.cc | 382 const int64 batch_size_; member in tensorflow::data::__anone2805d3c0111::PaddedBatchDatasetOp::Dataset
|
| /external/libchrome/mojo/public/cpp/bindings/tests/ |
| D | e2e_perftest.cc | 65 int batch_size_; member in mojo::__anonbc74313d0111::PingPongTest
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 308 const int64 batch_size_; member in tensorflow::data::__anon017c0f150111::DenseToSparseBatchDatasetOp::Dataset
|
| D | map_and_batch_dataset_op.cc | 812 const int64 batch_size_; member in tensorflow::data::__anon2c387ec70111::MapAndBatchDatasetOp::Dataset
|
| D | numa_map_and_batch_dataset_op.cc | 1139 const int64 batch_size_; member in tensorflow::data::__anon9271f8770111::NumaMapAndBatchDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 1717 const int batch_size_ = 4; member in tensorflow::grappler::VirtualSchedulerTest
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1141 int batch_size_; member in stream_executor::gpu::CudnnRnnDescriptor
1355 int batch_size_; member in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1393 int batch_size_; member in stream_executor::gpu::CudnnRnnStateTensorDescriptor
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1827 int batch_size_; member in stream_executor::gpu::MIOpenRnnSequenceTensorDescriptor
1872 int batch_size_; member in stream_executor::gpu::MIOpenRnnStateTensorDescriptor
|