Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.h63 const int64 batch_size_; variable
Dtriangular_solve_thunk.h63 const int64 batch_size_; variable
/external/tensorflow/tensorflow/core/kernels/
Drecord_input_op.cc66 int64 batch_size_; member in tensorflow::RecordInputOp
Dword2vec_kernels.cc112 int32 batch_size_ = 0; member in tensorflow::SkipgramOp
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Ddepth_to_space_tester.h87 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
Dconv_2d_tester.h211 int32_t batch_size_ = 1; variable
Ddepthwise_conv_2d_tester.h215 int32_t batch_size_ = 1; variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dunion_find.h113 absl::optional<int> batch_size_; variable
/external/tensorflow/tensorflow/lite/experimental/kernels/
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/
Dconvert_nodes.h346 int batch_size_ = -1; variable
677 int batch_size_ = -1; variable
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunbatch_dataset_op.cc247 int64 batch_size_; member in tensorflow::data::experimental::__anon91d1b6390111::UnbatchDatasetOp::Dataset
Dmap_and_batch_dataset_op_test.cc91 int64 batch_size_; member in tensorflow::data::experimental::__anon402561240111::MapAndBatchDatasetParams
Ddense_to_sparse_batch_dataset_op.cc317 const int64 batch_size_; member in tensorflow::data::experimental::__anone64ec8730111::DenseToSparseBatchDatasetOp::Dataset
Dmap_and_batch_dataset_op.cc681 const int64 batch_size_; member in tensorflow::data::experimental::MapAndBatchDatasetOp::Dataset
/external/libchrome/mojo/public/cpp/bindings/tests/
De2e_perftest.cc65 int batch_size_; member in mojo::__anonf408f29b0111::PingPongTest
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op_test.cc74 int64 batch_size_; member in tensorflow::data::__anon57c7b8840111::ParallelBatchDatasetParams
Dbatch_dataset_op.cc246 const int64 batch_size_; member in tensorflow::data::BatchDatasetOp::Dataset
Dpadded_batch_dataset_op.cc383 const int64 batch_size_; member in tensorflow::data::PaddedBatchDatasetOp::Dataset
Dpadded_batch_dataset_op_test.cc96 int64 batch_size_; member in tensorflow::data::__anon1aa214390111::PaddedBatchDatasetParams
Dparallel_batch_dataset_op.cc515 const int64 batch_size_; member in tensorflow::data::ParallelBatchDatasetOp::Dataset
/external/perfetto/src/trace_processor/sqlite/
Dsqlite_vtable_benchmark.cc65 size_t batch_size_; member in __anon036853120111::BenchmarkCursor
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc2393 const int batch_size_ = 4; member in tensorflow::grappler::__anon8eb856130111::VirtualSchedulerTest

12