| /external/tensorflow/tensorflow/core/kernels/ |
| dynamic_partition_op.cc | 34 OP_REQUIRES_OK(c, c->GetAttr("num_partitions", &num_partitions_));
56 gtl::InlinedVector<int, 32> partition_count(num_partitions_);
61 OP_REQUIRES(c, FastBoundsCheck(p, num_partitions_),
64 " = ", p, " is not in [0, ", num_partitions_, ")"));
70 for (int p = 0; p < num_partitions_; p++) {
82 int num_partitions_; member in class:tensorflow::DynamicPartitionOp_Shared
96 if (num_partitions_ == 0 || data->NumElements() == 0) return;
100 gtl::InlinedVector<int, 32> output_index(num_partitions_);
108 out_vec.reserve(num_partitions_);
109 for (int p = 0; p < num_partitions_; p++)
[all...] |
| dynamic_partition_op_gpu.cu.cc | 192 OP_REQUIRES_OK(c, c->GetAttr("num_partitions", &num_partitions_));
193 OP_REQUIRES(c, num_partitions_ >= 1,
200 int32 M = std::max(N, num_partitions_);
217 for (int p = 0; p < num_partitions_; p++) {
249 c->allocate_temp(DT_INT32, TensorShape({num_partitions_}),
253 for (int i = 0; i < num_partitions_; i++) e_part_count(i) = 0;
264 c->allocate_temp(DT_INT32, TensorShape({num_partitions_}),
290 num_partitions_ * sizeof(int32));
294 num_partitions_ * sizeof(int32))
384 c->allocate_temp(DT_INT32, TensorShape({num_partitions_}),
456 int32 num_partitions_; member in class:tensorflow::DynamicPartitionOpGPU
[all...] |
| /external/webrtc/webrtc/modules/rtp_rtcp/source/ |
| vp8_partition_aggregator.cc | 28 num_partitions_(num_partitions),
53 if (num_partitions_ == 0) {
66 if (num_partitions_ > 0) {
70 new PartitionTreeNode(this, &size_vector_[1], num_partitions_ - 1,
81 this, &size_vector_[1], num_partitions_ - 1, size_vector_[0]);
149 num_partitions_(last_partition_idx - first_partition_idx + 1),
150 size_vector_(new size_t[num_partitions_]),
154 for (size_t i = 0; i < num_partitions_; ++i) {
160 root_ = PartitionTreeNode::CreateRootNode(size_vector_, num_partitions_);
182 ConfigVec config_vector(num_partitions_, 0)
[all...] |
| vp8_partition_aggregator.h | 76 size_t num_partitions_; member in class:webrtc::PartitionTreeNode
129 size_t num_partitions_; member in class:webrtc::Vp8PartitionAggregator
|
| rtp_format_vp8.cc | 160 num_partitions_(0),
175 num_partitions_(0),
191 num_partitions_ = fragmentation->fragmentationVectorSize;
196 num_partitions_ = part_info_.fragmentationVectorSize;
311 if (part_ix + 1 < num_partitions_ &&
314 assert(part_ix < num_partitions_);
358 while (part_ix < num_partitions_) {
411 partition_vec->assign(num_partitions_, -1);
419 while (first_in_set < num_partitions_) {
423 while (last_in_set + 1 < num_partitions_ &
[all...] |
| rtp_format_vp8.h | 210 size_t num_partitions_; member in class:webrtc::RtpPacketizerVp8
|
| /external/tensorflow/tensorflow/contrib/cloud/kernels/ |
| bigquery_reader_ops.cc | 150 total_num_rows_, num_partitions_);
153 context->allocate_output(0, TensorShape({num_partitions_}),
157 for (int64 i = 0; i < num_partitions_; ++i) {
176 TF_RETURN_IF_ERROR(context->GetAttr("num_partitions", &num_partitions_));
177 if (num_partitions_ <= 0) {
183 int64 num_partitions_; member in class:tensorflow::GenerateBigQueryReaderPartitionsOp
|
