| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | triangular_solve_thunk.h | 68 const int64_t batch_size_; variable
|
| H A D | cholesky_thunk.h | 65 const int64_t batch_size_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | record_input_op.cc | 66 int64_t batch_size_; member in tensorflow::RecordInputOp
|
| /aosp_15_r20/system/core/fs_mgr/libsnapshot/libsnapshot_cow/ |
| H A D | writer_v3.h | 115 size_t batch_size_ = 1; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | depth_to_space_tester.h | 93 int32_t batch_size_ = 1; variable
|
| H A D | resize_bilinear_tester.h | 102 int32_t batch_size_ = 1; variable
|
| H A D | pool_2d_tester.h | 161 int32_t batch_size_ = 1; variable
|
| H A D | quantized_resize_bilinear_tester.h | 129 int32_t batch_size_ = 1; variable
|
| H A D | quantized_pool_2d_tester.h | 187 int32_t batch_size_ = 1; variable
|
| H A D | transpose_conv_tester.h | 216 int32_t batch_size_ = 1; variable
|
| H A D | depthwise_conv_2d_tester.h | 238 int32_t batch_size_ = 1; variable
|
| H A D | conv_2d_tester.h | 247 int32_t batch_size_ = 1; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| H A D | union_find.h | 113 std::optional<int> batch_size_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/ |
| H A D | ctc_decoder.h | 66 int batch_size_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ctc/ |
| H A D | ctc_decoder.h | 69 int batch_size_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | weights.h | 273 int batch_size_ = -1; variable
|
| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | vlrelu-microkernel-tester.h | 181 size_t batch_size_ = 1; variable
|
| H A D | vcvt-microkernel-tester.h | 405 size_t batch_size_ = 1; variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MultinomialTest.cpp | 102 const uint32_t batch_size_; member in android::nn::wrapper::MultinomialOpModel
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | unbatch_dataset_op.cc | 255 int64_t batch_size_; member in tensorflow::data::experimental::__anonfb83e4280111::UnbatchDatasetOp::Dataset
|
| H A D | dense_to_sparse_batch_dataset_op.cc | 317 const int64_t batch_size_; member in tensorflow::data::experimental::__anon5d7b2ca20111::DenseToSparseBatchDatasetOp::Dataset
|
| H A D | map_and_batch_dataset_op_test.cc | 91 int64_t batch_size_; member in tensorflow::data::experimental::__anon71359bb30111::MapAndBatchDatasetParams
|
| /aosp_15_r20/external/libchrome/mojo/public/cpp/bindings/tests/ |
| H A D | e2e_perftest.cc | 65 int batch_size_; member in mojo::__anon1e585aca0111::PingPongTest
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/data/datasets/ |
| H A D | chunk.h | 204 size_t batch_size_ = 0; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | parallel_batch_dataset_op_test.cc | 82 int64_t batch_size_; member in tensorflow::data::__anonfde464930111::ParallelBatchDatasetParams
|