Home
last modified time | relevance | path

Searched defs:batch_size_ (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Drecord_input_op.cc66 int64_t batch_size_; member in tensorflow::RecordInputOp
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtriangular_solve_thunk.h68 const int64_t batch_size_; variable
Dcholesky_thunk.h65 const int64_t batch_size_; variable
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Ddepth_to_space_tester.h93 int32_t batch_size_ = 1; variable
Dresize_bilinear_tester.h102 int32_t batch_size_ = 1; variable
Dpool_2d_tester.h161 int32_t batch_size_ = 1; variable
Dquantized_resize_bilinear_tester.h129 int32_t batch_size_ = 1; variable
Dquantized_pool_2d_tester.h187 int32_t batch_size_ = 1; variable
Ddepthwise_conv_2d_tester.h238 int32_t batch_size_ = 1; variable
Dtranspose_conv_tester.h216 int32_t batch_size_ = 1; variable
Dconv_2d_tester.h247 int32_t batch_size_ = 1; variable
Dquantized_depthwise_conv_2d_tester.h275 int32_t batch_size_ = 1; variable
Dquantized_conv_2d_tester.h276 int32_t batch_size_ = 1; variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dunion_find.h113 std::optional<int> batch_size_; variable
/external/tensorflow/tensorflow/lite/kernels/ctc/
Dctc_decoder.h66 int batch_size_; variable
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_decoder.h69 int batch_size_; variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dweights.h273 int batch_size_ = -1; variable
/external/XNNPACK/test/
Dvlrelu-microkernel-tester.h181 size_t batch_size_ = 1; variable
Dvcvt-microkernel-tester.h405 size_t batch_size_ = 1; variable
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunbatch_dataset_op.cc255 int64_t batch_size_; member in tensorflow::data::experimental::__anon3dae43170111::UnbatchDatasetOp::Dataset
Ddense_to_sparse_batch_dataset_op.cc317 const int64_t batch_size_; member in tensorflow::data::experimental::__anoncb2181d10111::DenseToSparseBatchDatasetOp::Dataset
Dmap_and_batch_dataset_op_test.cc91 int64_t batch_size_; member in tensorflow::data::experimental::__anond49d67420111::MapAndBatchDatasetParams
/external/libchrome/mojo/public/cpp/bindings/tests/
De2e_perftest.cc65 int batch_size_; member in mojo::__anon2b9db3f90111::PingPongTest
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op_test.cc82 int64_t batch_size_; member in tensorflow::data::__anon45b721a20111::ParallelBatchDatasetParams
Dbatch_dataset_op.cc277 const int64_t batch_size_; member in tensorflow::data::BatchDatasetOp::Dataset

12