| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | union_find.cc | 57 : batch_size_(absl::nullopt), max_batch_size_(absl::nullopt) {} in ClusterBatchSize()
60 return batch_size_ == other.batch_size_ && in operator ==()
71 batch_size_ = MergeCompatible<int>(batch_size_, batch_size); in SetBatchSize()
72 if (batch_size_.has_value() && batch_size_.value() >= 0) { in SetBatchSize()
73 SetMaxBatchSize(batch_size_); in SetBatchSize()
78 bool ClusterBatchSize::HasBatchSize() const { return batch_size_.has_value(); } in HasBatchSize()
82 return batch_size_.value(); in GetBatchSize()
101 if (!CheckIfCompatible(batch_size_, other.batch_size_) || in MergeIfCompatible()
106 SetBatchSize(other.batch_size_); in MergeIfCompatible()
121 append_optional_num(batch_size_); in ToString()
[all …]
|
| D | union_find.h | 113 absl::optional<int> batch_size_; variable
133 const ClusterBatchSize& BatchSize() const { return batch_size_; } in BatchSize()
142 ClusterBatchSize batch_size_;
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_decoder.h | 46 batch_size_(batch_size), in CTCDecoder()
60 int batch_size() { return batch_size_; } in batch_size()
66 int batch_size_; variable
81 if (output->empty() || (*output)[0].size() < batch_size_) { in Decode()
84 if (scores->rows() < batch_size_ || scores->cols() == 0) { in Decode()
88 for (int b = 0; b < batch_size_; ++b) { in Decode()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_decoder.h | 49 batch_size_(batch_size), in CTCDecoder()
63 int batch_size() { return batch_size_; } in batch_size()
69 int batch_size_; variable
86 if (output->empty() || (*output)[0].size() < Decoder::batch_size_) { in Decode()
90 if (scores->rows() < Decoder::batch_size_ || scores->cols() == 0) { in Decode()
95 for (int b = 0; b < Decoder::batch_size_; ++b) { in Decode()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 53 batch_size_(batch_size), in Dataset()
79 PartialTensorShape({batch_size_}).Concatenate(input_shape));
108 params.set_args(batch_size_); in DebugString()
117 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
136 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size)); in AsGraphDefInternal()
171 for (int i = 0; i < dataset()->batch_size_ && !*end_of_sequence; ++i) { in GetNextInternal()
189 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
212 return model::MakeKnownRatioNode(std::move(args), dataset()->batch_size_); in CreateNode()
246 const int64 batch_size_; member in tensorflow::data::BatchDatasetOp::Dataset
|
| D | padded_batch_dataset_op.cc | 56 batch_size_(batch_size), in Dataset()
81 PartialTensorShape({batch_size_}).Concatenate(padded_shapes_[i]));
110 params.set_args(batch_size_); in DebugString()
119 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
138 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size)); in AsGraphDefInternal()
205 batch_elements.reserve(dataset()->batch_size_); in GetNextInternal()
206 for (int i = 0; i < dataset()->batch_size_ && !*end_of_sequence; in GetNextInternal()
227 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
348 return model::MakeKnownRatioNode(std::move(args), dataset()->batch_size_); in CreateNode()
383 const int64 batch_size_; member in tensorflow::data::PaddedBatchDatasetOp::Dataset
|
| D | parallel_batch_dataset_op.cc | 69 batch_size_(batch_size), in Dataset()
92 PartialTensorShape({batch_size_}).Concatenate(input_shape));
118 params.set_args(batch_size_); in DebugString()
127 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
149 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size)); in AsGraphDefInternal()
225 ProcessBatch(dataset()->batch_size_, result->num_elements, in GetNextInternal()
236 /*ratio=*/dataset()->batch_size_, /*memory_ratio=*/1.0, in CreateNode()
337 for (int i = 0; i < dataset()->batch_size_ && !end_of_input; ++i) { in CallBatching()
453 TF_RETURN_IF_ERROR(ReadBatch(dataset()->batch_size_, prefix(), in ReadBatchResult()
479 TF_RETURN_IF_ERROR(WriteBatch(dataset()->batch_size_, in WriteBatchResult()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | record_input_op.cc | 53 batch_size_ = batch_size; in RecordInputOp()
57 Tensor out(DT_STRING, {batch_size_}); in Compute()
59 for (int i = 0; i < batch_size_; ++i) { in Compute()
66 int64 batch_size_; member in tensorflow::RecordInputOp
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 82 batch_size_(batch_size), in Dataset()
111 return strings::StrCat("DenseToSparseBatchDatasetOp(", batch_size_, in DebugString()
120 return n / batch_size_ + (n % batch_size_ == 0 ? 0 : 1); in Cardinality()
140 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
172 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in GetNextInternal()
193 i < DatasetIterator<Dataset<T>>::dataset()->batch_size_ && in GetNextInternal()
295 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in CreateNode()
317 const int64 batch_size_; member in tensorflow::data::experimental::__anone64ec8730111::DenseToSparseBatchDatasetOp::Dataset
|
| D | map_and_batch_dataset_op.cc | 92 batch_size_(batch_size), in Dataset()
134 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
154 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
208 params.dataset->batch_size_)); in Iterator()
263 ProcessBatch(dataset()->batch_size_, result->num_elements, in GetNextInternal()
273 std::move(args), dataset()->batch_size_, in CreateNode()
508 TensorShape component_shape({dataset()->batch_size_}); in EnsureOutputAllocated()
540 call_counter_ % dataset()->batch_size_ == 0)); in RunnerThread()
565 if (call_counter_ % dataset()->batch_size_ == 0) { in RunnerThread()
567 std::make_shared<BatchResult>(dataset()->batch_size_)); in RunnerThread()
[all …]
|
| D | unbatch_dataset_op.cc | 41 batch_size_ = -1; in Dataset()
44 if (batch_size_ < 0 && shape.dim_size(0) >= 0) { in Dataset()
45 batch_size_ = shape.dim_size(0); in Dataset()
81 if (batch_size_ > 0) { in Cardinality()
82 return n * batch_size_; in Cardinality()
247 int64 batch_size_; member in tensorflow::data::experimental::__anon91d1b6390111::UnbatchDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | depth_to_space_tester.h | 36 batch_size_ = batch_size; in BatchSize()
40 inline int32_t BatchSize() const { return batch_size_; } in BatchSize()
87 int32_t batch_size_ = 1; variable
|
| D | resize_bilinear_tester.h | 37 batch_size_ = batch_size; in BatchSize()
41 inline int32_t BatchSize() const { return batch_size_; } in BatchSize()
102 int32_t batch_size_ = 1; variable
|
| D | pool_2d_tester.h | 37 batch_size_ = batch_size; in BatchSize()
41 inline int32_t BatchSize() const { return batch_size_; } in BatchSize()
161 int32_t batch_size_ = 1; variable
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | gru_ops.h | 32 : batch_size_(batch_size), in GRUCell()
39 return {batch_size_, input_size_}; in x_extends()
47 return {batch_size_, cell_size_}; in h_extends()
59 return {batch_size_, cell_size_}; in cell_extents()
63 const int batch_size_;
96 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({batch_size_, 1}); in operator()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_int8_calibrator.cc | 30 int TRTInt8Calibrator::getBatchSize() const { return batch_size_; } in getBatchSize()
35 : batch_size_(batch_size), in TRTInt8Calibrator()
44 : batch_size_(0), in TRTInt8Calibrator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cholesky_thunk.cc | 47 batch_size_(batch_size), in CholeskyThunk()
54 << " batch_size=" << batch_size_ << " n=" << n_ in ExecuteOnStream()
76 for (int64 i = 0; i < batch_size_; ++i) { in ExecuteOnStream()
|
| /external/libchrome/mojo/public/cpp/bindings/tests/ |
| D | e2e_perftest.cc | 65 int batch_size_; member in mojo::__anonf408f29b0111::PingPongTest
81 batch_size_ = batch_size; in RunTest()
102 calls_outstanding_ = batch_size_; in DoPing()
103 for (int i = 0; i < batch_size_; i++) { in DoPing()
|
| /external/perfetto/src/trace_processor/sqlite/ |
| D | sqlite_vtable_benchmark.cc | 52 : num_cols_(num_cols), batch_size_(batch_size), rnd_engine_(kRandomSeed) { in BenchmarkCursor()
65 size_t batch_size_; member in __anon036853120111::BenchmarkCursor
77 for (size_t row = 0; row < batch_size_; row++) { in RandomFill()
84 row_ = (row_ + 1) % batch_size_; in Next()
|
| /external/XNNPACK/test/ |
| D | copy-operator-tester.h | 66 this->batch_size_ = batch_size; in batch_size()
71 return this->batch_size_; in batch_size()
137 size_t batch_size_{1};
|
| D | negate-operator-tester.h | 65 this->batch_size_ = batch_size; in batch_size()
70 return this->batch_size_; in batch_size()
136 size_t batch_size_{1};
|
| D | ceiling-operator-tester.h | 66 this->batch_size_ = batch_size; in batch_size()
71 return this->batch_size_; in batch_size()
137 size_t batch_size_{1};
|
| D | square-operator-tester.h | 65 this->batch_size_ = batch_size; in batch_size()
70 return this->batch_size_; in batch_size()
137 size_t batch_size_{1};
|
| D | square-root-operator-tester.h | 65 this->batch_size_ = batch_size; in batch_size()
70 return this->batch_size_; in batch_size()
137 size_t batch_size_{1};
|
| D | truncation-operator-tester.h | 66 this->batch_size_ = batch_size; in batch_size()
71 return this->batch_size_; in batch_size()
137 size_t batch_size_{1};
|