| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | dynamic_partition_op_test.py | 44 data, indices, num_partitions=4)
67 data, indices, num_partitions=4)
92 data, indices, num_partitions=2)
105 num_partitions = 97
106 indices_list = [(i ** 2) % num_partitions for i in range(rows)]
107 parts = [[] for _ in range(num_partitions)]
109 parts[(i ** 2) % num_partitions].append(data_list[i])
114 data, indices, num_partitions=num_partitions)
117 self.assertEqual(num_partitions, len(partition_vals))
118 for i in range(num_partitions):
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unpartition_embedding_lookup.cc | 60 for (size_t i = 0; i < stitch_op->num_partitions; ++i) { in Run()
63 for (size_t i = stitch_op->num_partitions; i < stitch_op->num_partitions * 2; in Run()
156 CHECK_EQ(indices_partition_op->num_partitions, in Run()
157 data_partition_op->num_partitions) in Run()
159 int num_partitions = indices_partition_op->num_partitions; in Run() local
204 {num_partitions * partition_array_dims[0]}); in Run()
208 for (int i = 0; i < num_partitions * partition_array_dims[0]; ++i) { in Run()
209 int p = i % num_partitions; in Run()
210 perm_data[i] = p * partition_array_dims[0] + i / num_partitions; in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fork_join.cc | 64 void** buffer_table, uint64* prof_counters, int32 num_partitions, in __xla_cpu_runtime_ParallelForkJoin() argument
67 << " num_partitions: " << num_partitions in __xla_cpu_runtime_ParallelForkJoin()
70 CHECK_GT(num_partitions, 1); in __xla_cpu_runtime_ParallelForkJoin()
85 tensorflow::BlockingCounter bc(num_partitions - 1); in __xla_cpu_runtime_ParallelForkJoin()
86 for (int32 i = 1; i < num_partitions; ++i) { in __xla_cpu_runtime_ParallelForkJoin()
|
| D | ir_function.cc | 271 const int64 num_partitions = partition_iterator.GetTotalPartitionCount(); in EmitCallToParallelForkJoin() local
273 fork_join_arguments.push_back(b->getInt32(num_partitions)); in EmitCallToParallelForkJoin()
284 dim_partition_size * num_partitioned_dims * num_partitions; in EmitCallToParallelForkJoin()
289 for (int32 i = 0; i < num_partitions; ++i) { in EmitCallToParallelForkJoin()
|
| D | runtime_fork_join.h | 28 tensorflow::int32 num_partitions, tensorflow::int64* partitions,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DynamicPartition.pbtxt | 6 Any shape. Indices in the range `[0, num_partitions)`.
10 name: "num_partitions"
15 summary: "Partitions `data` into `num_partitions` tensors using indices from `partitions`."
36 num_partitions = 2
43 num_partitions = 2
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 299 def _get_first_dimension_size_statically(self, w, num_partitions): argument
302 for p in range(1, num_partitions):
370 num_partitions): argument
372 num_partitions = num_partitions_by_var[v_num]
376 full_update, p_assignments, num_partitions)
378 for p in range(num_partitions):
434 num_partitions = len(w)
441 w, num_partitions)
449 for p in range(num_partitions):
457 ids_per_partition = num_total_ids // num_partitions
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_test.cc | 162 Node* DynamicPartitionNode(Graph* g, Node* in0, Node* in1, int num_partitions) { in DynamicPartitionNode() argument
167 .Attr("num_partitions", num_partitions) in DynamicPartitionNode()
173 static Graph* DynamicPartition(int num_partitions, int dim) { in DynamicPartition() argument
184 partitions.flat<int32>()(i) = rnd.Uniform(num_partitions); in DynamicPartition()
187 test::graph::Constant(g, partitions), num_partitions); in DynamicPartition()
|
| D | smooth-hinge-loss.h | 33 double ComputeUpdatedDual(const int num_partitions, const double label, in ComputeUpdatedDual() argument
46 (num_partitions * example_weight * weighted_example_norm + gamma); in ComputeUpdatedDual()
|
| /external/webrtc/webrtc/modules/audio_processing/aec/ |
| D | aec_core_mips.c | 326 int num_partitions, in WebRtcAec_FilterFar_mips() argument
332 for (i = 0; i < num_partitions; i++) { in WebRtcAec_FilterFar_mips()
336 if (i + x_fft_buf_block_pos >= num_partitions) { in WebRtcAec_FilterFar_mips()
337 xPos -= num_partitions * (PART_LEN1); in WebRtcAec_FilterFar_mips()
443 int num_partitions, in WebRtcAec_FilterAdaptation_mips() argument
450 for (i = 0; i < num_partitions; i++) { in WebRtcAec_FilterAdaptation_mips()
454 if (i + x_fft_buf_block_pos >= num_partitions) { in WebRtcAec_FilterAdaptation_mips()
455 xPos -= num_partitions * PART_LEN1; in WebRtcAec_FilterAdaptation_mips()
|
| D | aec_core.c | 157 int num_partitions, in FilterFar() argument
163 for (i = 0; i < num_partitions; i++) { in FilterFar()
168 if (i + x_fft_buf_block_pos >= num_partitions) { in FilterFar()
169 xPos -= num_partitions * (PART_LEN1); in FilterFar()
215 int num_partitions, in FilterAdaptation() argument
222 for (i = 0; i < num_partitions; i++) { in FilterAdaptation()
226 if (i + x_fft_buf_block_pos >= num_partitions) { in FilterAdaptation()
227 xPos -= num_partitions * PART_LEN1; in FilterAdaptation()
302 for (i = 0; i < aec->num_partitions; i++) { in PartitionDelay()
801 for (i = lookahead; i < lookahead + self->num_partitions; ++i) { in UpdateDelayMetrics()
[all …]
|
| D | aec_core_internal.h | 153 int num_partitions; member
177 int num_partitions,
190 int num_partitions,
|
| D | aec_core_neon.c | 38 int num_partitions, in FilterFarNEON() argument
44 for (i = 0; i < num_partitions; i++) { in FilterFarNEON()
49 if (i + x_fft_buf_block_pos >= num_partitions) { in FilterFarNEON()
50 xPos -= num_partitions * PART_LEN1; in FilterFarNEON()
190 int num_partitions, in FilterAdaptationNEON() argument
197 for (i = 0; i < num_partitions; i++) { in FilterAdaptationNEON()
202 if (i + x_fft_buf_block_pos >= num_partitions) { in FilterAdaptationNEON()
203 xPos -= num_partitions * PART_LEN1; in FilterAdaptationNEON()
465 for (i = 0; i < aec->num_partitions; i++) { in PartitionDelayNEON()
|
| D | aec_core_sse2.c | 33 int num_partitions, in FilterFarSSE2() argument
40 for (i = 0; i < num_partitions; i++) { in FilterFarSSE2()
45 if (i + x_fft_buf_block_pos >= num_partitions) { in FilterFarSSE2()
46 xPos -= num_partitions * (PART_LEN1); in FilterFarSSE2()
152 int num_partitions, in FilterAdaptationSSE2() argument
159 for (i = 0; i < num_partitions; i++) { in FilterAdaptationSSE2()
163 if (i + x_fft_buf_block_pos >= num_partitions) { in FilterAdaptationSSE2()
164 xPos -= num_partitions * PART_LEN1; in FilterAdaptationSSE2()
452 for (i = 0; i < aec->num_partitions; i++) { in PartitionDelaySSE2()
|
| /external/tensorflow/tensorflow/contrib/cloud/python/ops/ |
| D | bigquery_reader_ops.py | 72 num_partitions, argument
114 self._num_partitions = num_partitions
145 num_partitions=self._num_partitions,
|
| D | bigquery_reader_ops_test.py | 213 num_partitions=4,
262 num_partitions=4,
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | data_flow_grad.cc | 66 int32 num_partitions; in DynamicPartitionGrad() local
68 GetNodeAttr(op.node()->attrs(), "num_partitions", &num_partitions)); in DynamicPartitionGrad()
90 DynamicPartition(scope, original_indices, partitions, num_partitions); in DynamicPartitionGrad()
|
| /external/webrtc/webrtc/modules/rtp_rtcp/source/ |
| D | rtp_format_vp8_test_helper.cc | 39 size_t num_partitions) { in Init() argument
42 fragmentation_->VerifyAndAllocateFragmentationHeader(num_partitions); in Init()
45 for (size_t p = 0; p < num_partitions; ++p) { in Init()
53 for (size_t p = 0; p < num_partitions; ++p) { in Init()
|
| D | vp8_partition_aggregator.cc | 23 size_t num_partitions, in PartitionTreeNode() argument
28 num_partitions_(num_partitions), in PartitionTreeNode()
39 size_t num_partitions) { in CreateRootNode() argument
41 NULL, &size_vector[1], num_partitions - 1, size_vector[0]); in CreateRootNode()
|
| D | vp8_partition_aggregator.h | 28 size_t num_partitions,
33 size_t num_partitions);
|
| D | rtp_format_vp8_test_helper.h | 34 bool Init(const size_t* partition_sizes, size_t num_partitions);
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_grad.py | 35 num_partitions = op.get_attr("num_partitions")
41 original_indices, indices, num_partitions)
|
| /external/vboot_reference/host/include/ |
| D | cgpt_params.h | 55 int num_partitions; member
|
| /external/kernel-headers/original/uapi/linux/ |
| D | switchtec_ioctl.h | 39 __u32 num_partitions; member
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 347 num_partitions = len(params)
349 for p in range(num_partitions):
357 for p in range(num_partitions):
358 expected_size = (num_params - p - 1) // num_partitions + 1
|