Home
last modified time | relevance | path

Searched refs:batch_size_ (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dbatch_dataset_op.cc57 batch_size_(batch_size), in Dataset()
70 PartialTensorShape({batch_size_}).Concatenate(input_shape)); in Dataset()
95 return strings::StrCat("BatchDatasetOp(", batch_size_, ")::Dataset"); in DebugString()
103 return n / batch_size_ + in Cardinality()
104 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
114 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size)); in AsGraphDefInternal()
144 batch_elements.reserve(dataset()->batch_size_); in GetNextInternal()
146 for (int i = 0; i < dataset()->batch_size_ && !*end_of_sequence; in GetNextInternal()
165 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
218 dataset()->batch_size_); in CreateNode()
[all …]
Dpadded_batch_dataset_op.cc107 batch_size_(batch_size), in Dataset()
126 PartialTensorShape({batch_size_}).Concatenate(padded_shapes_[i])); in Dataset()
151 return strings::StrCat("PaddedBatchDatasetOp(", batch_size_, in DebugString()
160 return n / batch_size_ + in Cardinality()
161 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
171 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size)); in AsGraphDefInternal()
235 batch_elements.reserve(dataset()->batch_size_); in GetNextInternal()
236 for (int i = 0; i < dataset()->batch_size_ && !*end_of_sequence; in GetNextInternal()
257 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
352 dataset()->batch_size_); in CreateNode()
[all …]
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_decoder.h46 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()
Dctc_beam_search.h167 return output.size() < this->batch_size_; in Decode()
171 if (scores->rows() < batch_size_ || scores->cols() < top_n) { in Decode()
175 for (int b = 0; b < batch_size_; ++b) { in Decode()
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_decoder.h46 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()
85 if (scores->rows() < batch_size_ || scores->cols() == 0) { in Decode()
90 for (int b = 0; b < batch_size_; ++b) { in Decode()
Dctc_beam_search.h168 return output.size() < this->batch_size_; in Decode()
173 if (scores->rows() < batch_size_ || scores->cols() < top_n) { in Decode()
178 for (int b = 0; b < batch_size_; ++b) { in Decode()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dbatch_features.h33 explicit BatchFeatures(int64 batch_size) : batch_size_(batch_size) {} in BatchFeatures()
68 QCHECK(example_start < batch_size_ && example_end <= batch_size_); in examples_iterable()
75 int64 batch_size() const { return batch_size_; } in batch_size()
79 const int64 batch_size_;
Dbatch_features.cc51 dense_float_feature.dim_size(0) == batch_size_, in Initialize()
53 "Dense float vector must have batch_size rows: ", batch_size_, in Initialize()
95 shape_flat(0) == batch_size_, in Initialize()
137 shape_flat(0) == batch_size_, in Initialize()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc84 batch_size_(batch_size), in Dataset()
113 return strings::StrCat("DenseToSparseBatchDatasetOp(", batch_size_, in DebugString()
122 return n / batch_size_ + (n % batch_size_ == 0 ? 0 : 1); in Cardinality()
132 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
164 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in GetNextInternal()
185 i < DatasetIterator<Dataset<T>>::dataset()->batch_size_ && in GetNextInternal()
287 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in CreateNode()
308 const int64 batch_size_; member in tensorflow::data::__anon017c0f150111::DenseToSparseBatchDatasetOp::Dataset
Dmap_and_batch_dataset_op.cc157 batch_size_(batch_size), in Dataset()
195 return n / batch_size_ + in Cardinality()
196 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
207 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
265 params.dataset->batch_size_ - 1) / in Iterator()
266 params.dataset->batch_size_)) { in Iterator()
317 std::move(args), dataset()->batch_size_, in CreateNode()
533 TensorShape component_shape({dataset()->batch_size_}); in EnsureOutputAllocated()
569 if (result->num_elements < dataset()->batch_size_) { in ProcessResult()
611 call_counter_ % dataset()->batch_size_ == 0)); in RunnerThread()
[all …]
Dnuma_map_and_batch_dataset_op.cc109 batch_size_(batch_size), in Dataset()
146 return n / batch_size_ + in Cardinality()
147 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
158 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
265 std::move(args), dataset()->batch_size_, in CreateNode()
649 itr_->dataset()->batch_size_); in Save()
696 error_index_int > itr->dataset()->batch_size_) { in Restore()
994 for (size_t i = 0; i < dataset()->batch_size_; ++i) { in RunnerThread()
1139 const int64 batch_size_; member in tensorflow::data::__anon9271f8770111::NumaMapAndBatchDatasetOp::Dataset
/external/tensorflow/tensorflow/core/kernels/
Drecord_input_op.cc53 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
Dword2vec_kernels.cc55 OP_REQUIRES_OK(ctx, ctx->GetAttr("batch_size", &batch_size_)); in SkipgramOp()
75 Tensor examples(DT_INT32, TensorShape({batch_size_})); in Compute()
77 Tensor labels(DT_INT32, TensorShape({batch_size_})); in Compute()
81 for (int i = 0; i < batch_size_; ++i) { in Compute()
112 int32 batch_size_ = 0; member in tensorflow::SkipgramOp
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dgru_ops.h32 : 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()
Dlstm_ops.h96 : batch_size_(batch_size), in LSTMBlockCell()
100 int batch_size() const { return batch_size_; } in batch_size()
123 return {batch_size_, cell_size_}; in cell_extents()
131 return {batch_size_, input_size_}; in xh_x_extents()
139 return {batch_size_, cell_size_}; in xh_h_extents()
143 const int batch_size_;
234 Eigen::array<Eigen::DenseIndex, 2> p_broadcast_shape({batch_size_, 1}); in operator()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_int8_calibrator.cc31 int TRTInt8Calibrator::getBatchSize() const { return batch_size_; } in getBatchSize()
36 : batch_size_(batch_size), in TRTInt8Calibrator()
45 : batch_size_(0), in TRTInt8Calibrator()
Dtrt_int8_calibrator.h78 const int batch_size_;
/external/libchrome/mojo/public/cpp/bindings/tests/
De2e_perftest.cc65 int batch_size_; member in mojo::__anonbc74313d0111::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/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc47 batch_size_(batch_size), in CholeskyThunk()
58 << " batch_size=" << batch_size_ << " n=" << n_ in ExecuteOnStream()
80 for (int64 i = 0; i < batch_size_; ++i) { in ExecuteOnStream()
Dtriangular_solve_thunk.cc50 batch_size_(batch_size), in TriangularSolveThunk()
77 << " batch_size=" << batch_size_ << " m=" << m_ << " n=" << n_ in ExecuteOnStream()
88 for (int64 i = 0; i < batch_size_; ++i) { in ExecuteOnStream()
Dcholesky_thunk.h66 const int64 batch_size_; variable
Dtriangular_solve_thunk.h65 const int64 batch_size_; variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h298 int batch_size() const { return batch_size_; } in batch_size()
305 void set_batch_size(int batch_size) { batch_size_ = batch_size; } in set_batch_size()
335 int batch_size_ = -1; variable
579 int batch_size_ = -1; variable
Dconvert_nodes.cc703 batch_size_(batch_size), in TRT_TensorOrWeights()
711 batch_size_(batch_size), in TRT_TensorOrWeights()
721 batch_size_(rhs.batch_size_), in TRT_TensorOrWeights()
729 batch_size_ = rhs.batch_size_; in operator =()
757 ", batch_size=", batch_size_); in DebugString()
1120 if (this->batch_size_ < 0 || batch_size < 0 || in MaybeUpdateBatchSize()
1121 this->batch_size_ == batch_size) { in MaybeUpdateBatchSize()
1122 if (this->batch_size_ < 0 && batch_size >= 0) { in MaybeUpdateBatchSize()
1123 this->batch_size_ = batch_size; in MaybeUpdateBatchSize()
1129 " vs ", batch_size_); in MaybeUpdateBatchSize()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc119 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds()
121 s.WithOpName("y"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds()
123 s.WithOpName("z"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2Ds()
146 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithConv2DAndVariable()
261 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithBatchNorm()
1549 s.WithOpName("x"), {batch_size_, width_, height_, depth_in_}, DT_FLOAT); in CreateGrapplerItemWithInterDeviceTransfers()
1717 const int batch_size_ = 4; member in tensorflow::grappler::VirtualSchedulerTest
2460 const int x_size = batch_size_ * width_ * height_ * depth_in_; in TEST_F()
2825 int input_size = 4 * batch_size_ * width_ * height_ * depth_in_; in TEST_F()

12