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