/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_dynamic_partition_op_test.py | 43 num_partitions=0, 49 num_partitions=3, 54 num_partitions=5, 60 num_partitions=4, 67 num_partitions=4, 74 num_partitions=5, 80 num_partitions=5, 86 num_partitions=2, 93 num_partitions=3, 100 num_partitions=3, [all …]
|
D | ragged_array_ops.py | 547 def stack_dynamic_partitions(data, partitions, num_partitions, name=None): argument 584 with ops.name_scope(name, 'SegmentStack', [data, partitions, num_partitions]): 592 num_partitions = ops.convert_to_tensor( 593 num_partitions, name='num_partitions', preferred_dtype=partitions.dtype) 596 num_partitions = math_ops.cast(num_partitions, partitions.dtype) 602 num_partitions.shape.assert_has_rank(0) 611 nrows=num_partitions, 623 num_partitions, 627 values, value_rowids, nrows=num_partitions, validate=False) 645 num_partitions)
|
/external/webrtc/modules/audio_processing/aec3/ |
D | adaptive_fir_filter_unittest.cc | 68 for (size_t num_partitions : {2, 5, 12, 30, 50}) { in TEST_P() 85 num_partitions, std::vector<FftData>(num_render_channels)); in TEST_P() 87 num_partitions, std::vector<FftData>(num_render_channels)); in TEST_P() 88 for (size_t p = 0; p < num_partitions; ++p) { in TEST_P() 118 AdaptPartitions_Neon(*render_buffer, G, num_partitions, &H_Neon); in TEST_P() 119 AdaptPartitions(*render_buffer, G, num_partitions, &H_C); in TEST_P() 120 AdaptPartitions_Neon(*render_buffer, G, num_partitions, &H_Neon); in TEST_P() 121 AdaptPartitions(*render_buffer, G, num_partitions, &H_C); in TEST_P() 123 for (size_t p = 0; p < num_partitions; ++p) { in TEST_P() 132 ApplyFilter_Neon(*render_buffer, num_partitions, H_Neon, &S_Neon); in TEST_P() [all …]
|
D | adaptive_fir_filter.h | 31 size_t num_partitions, 36 size_t num_partitions, 42 size_t num_partitions, 50 size_t num_partitions, 55 size_t num_partitions, 61 size_t num_partitions, 67 size_t num_partitions, 72 size_t num_partitions, 78 size_t num_partitions, 140 void SetFilter(size_t num_partitions,
|
D | adaptive_fir_filter.cc | 36 size_t num_partitions, in ComputeFrequencyResponse() argument 45 for (size_t p = 0; p < num_partitions; ++p) { in ComputeFrequencyResponse() 60 size_t num_partitions, in ComputeFrequencyResponse_Neon() argument 69 for (size_t p = 0; p < num_partitions; ++p) { in ComputeFrequencyResponse_Neon() 92 size_t num_partitions, in ComputeFrequencyResponse_Sse2() argument 102 for (size_t p = 0; p < num_partitions; ++p) { in ComputeFrequencyResponse_Sse2() 126 size_t num_partitions, in AdaptPartitions() argument 132 for (size_t p = 0; p < num_partitions; ++p) { in AdaptPartitions() 149 size_t num_partitions, in AdaptPartitions_Neon() argument 155 render_buffer_data.size() - render_buffer.Position(), num_partitions); in AdaptPartitions_Neon() [all …]
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | dynamic_partition_op_test.py | 46 data, indices, num_partitions=4) 69 data, indices, num_partitions=4) 94 data, indices, num_partitions=2) 107 num_partitions = 97 108 indices_list = [(i ** 2) % num_partitions for i in range(rows)] 109 parts = [[] for _ in range(num_partitions)] 111 parts[(i ** 2) % num_partitions].append(data_list[i]) 116 data, indices, num_partitions=num_partitions) 119 self.assertEqual(num_partitions, len(partition_vals)) 120 for i in range(num_partitions): [all …]
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | unpartition_embedding_lookup.cc | 60 stitch_indices_inputs.reserve(stitch_op->num_partitions); in Run() 61 for (int i = 0; i < stitch_op->num_partitions; ++i) { in Run() 64 for (int i = stitch_op->num_partitions; i < stitch_op->num_partitions * 2; in Run() 158 CHECK_EQ(indices_partition_op->num_partitions, in Run() 159 data_partition_op->num_partitions) in Run() 161 int num_partitions = indices_partition_op->num_partitions; in Run() local 206 {num_partitions * partition_array_dims[0]}); in Run() 210 for (int i = 0; i < num_partitions * partition_array_dims[0]; ++i) { in Run() 211 int p = i % num_partitions; in Run() 212 perm_data[i] = p * partition_array_dims[0] + i / num_partitions; in Run()
|
/external/tensorflow/tensorflow/python/distribute/ |
D | parameter_server_strategy_v2.py | 597 num_partitions = self._variable_partitioner(shape=shape, dtype=dtype) 598 if not num_partitions or num_partitions[0] == 0 or any( 599 v != 1 for v in num_partitions[1:]): 602 " besides the first element (non-zero), got: %r" % num_partitions) 604 if num_partitions[0] == 1: # no partition 608 num_partitions = min(num_partitions[0], shape[0]) 609 base = shape[0] // num_partitions 610 extra = shape[0] % num_partitions 614 for i in range(num_partitions): 663 for i in range(num_partitions):
|
/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()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | fft_handler.cc | 53 HloInstruction* hlo, int64 num_partitions, const HloSharding& sharding, in PadEachPartitionWithHaloExchange() argument 58 CeilOfRatio(size_per_partition, num_partitions) * num_partitions; in PadEachPartitionWithHaloExchange() 113 int64 num_partitions, in ShuffleWithinEachPartitionUsingOneHot() argument 116 CHECK_EQ(size_per_partition % num_partitions, 0); in ShuffleWithinEachPartitionUsingOneHot() 121 S32, {size_per_partition / num_partitions, num_partitions}), in ShuffleWithinEachPartitionUsingOneHot() 126 S32, {num_partitions, size_per_partition / num_partitions}), in ShuffleWithinEachPartitionUsingOneHot() 160 HloInstruction* hlo, int64 num_partitions, in ShuffleDataWithAllToAll() argument 164 std::vector<int64> partition_subgroups(num_partitions); in ShuffleDataWithAllToAll() 172 HloInstruction* GetCorrectionFactor(HloInstruction* hlo, int64 num_partitions, in GetCorrectionFactor() argument 188 complex64(0, -2.0 * M_PI / (num_partitions * per_replica_size))))); in GetCorrectionFactor() [all …]
|
D | dot_handler.cc | 327 int64 FirstShardingDimWithPartitionOfSize(int64 num_partitions, in FirstShardingDimWithPartitionOfSize() argument 331 if (sharding.tile_assignment().dim(i) == num_partitions) { in FirstShardingDimWithPartitionOfSize() 384 int64 num_partitions, int64 output_lhs_non_contracting_partitions, in GetWindowedEinsumConfiguration() argument 394 if (output_lhs_non_contracting_partitions == num_partitions && in GetWindowedEinsumConfiguration() 397 if (rhs_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration() 404 if (rhs_non_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration() 411 if (rhs_batch_partitions == num_partitions) { in GetWindowedEinsumConfiguration() 419 if (output_rhs_non_contracting_partitions == num_partitions && in GetWindowedEinsumConfiguration() 422 if (lhs_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration() 429 if (lhs_non_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration() [all …]
|
/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/python/keras/integration_test/ |
D | gradient_checkpoint_test.py | 29 def _get_big_cnn_model(img_dim, n_channels, num_partitions, argument 34 for _ in range(num_partitions): 48 def _get_split_cnn_model(img_dim, n_channels, num_partitions, argument 51 models = [tf.keras.Sequential() for _ in range(num_partitions)] 53 for i in range(num_partitions): 99 img_dim, n_channels, num_partitions=3, blocks_per_partition=2) 120 img_dim, n_channels, num_partitions=3, blocks_per_partition=2)
|
/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
D | tpu_client.cc | 83 int num_replicas, int num_partitions) const override; 101 int num_replicas, int num_partitions) const { in GetDefaultDeviceAssignment() 106 if (num_replicas * num_partitions <= num_local_devices) { in GetDefaultDeviceAssignment() 108 num_partitions); in GetDefaultDeviceAssignment() 112 num_partitions); in GetDefaultDeviceAssignment() 123 if (executable.num_partitions() > 1) { in ExecutableFingerprint()
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | sorting.cc | 41 int64 num_partitions = in TopK() local 43 if (num_partitions >= kMinNumPartitions) { in TopK() 44 return TopKWithPartitions(input, k, num_partitions); in TopK() 76 XlaOp TopKWithPartitions(XlaOp input, int64 k, int64 num_partitions) { in TopKWithPartitions() argument 84 const int64 per_partition_size = CeilOfRatio(last_dim_size, num_partitions); in TopKWithPartitions() 168 ForEachIndex(num_partitions - 1, S32, topk_body_fn, in TopKWithPartitions()
|
/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/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/tensorflow/tensorflow/compiler/xla/client/ |
D | executable_build_options.cc | 69 int num_partitions) { in set_num_partitions() argument 70 num_partitions_ = num_partitions; in set_num_partitions() 126 execution_options.set_num_partitions(build_options.num_partitions()); in CreateExecutionOptions()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | model_builder_test.cc | 316 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument 330 *num_partitions = interpreter_fp16_add_op->num_delegate_params(); in TEST() 375 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument 390 *num_partitions = interpreter_fp16_non_constant->num_delegate_params(); in TEST() 438 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument 442 *num_partitions = 0; in TEST() 552 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument 563 *num_partitions = interpreter_fp32->num_delegate_params(); in TEST() 719 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument 739 *num_partitions = interpreter2_fp32->num_delegate_params(); in TEST() [all …]
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
D | DynamicPartition.pbtxt | 14 number_attr: "num_partitions" 17 name: "num_partitions"
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | DynamicPartition.pbtxt | 14 number_attr: "num_partitions" 17 name: "num_partitions"
|
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
D | tpu_client.cc | 134 int num_replicas, int num_partitions) const { in GetDefaultDeviceAssignment() 135 if (num_partitions > 1) { in GetDefaultDeviceAssignment() 138 if (num_replicas * num_partitions <= local_device_count()) { in GetDefaultDeviceAssignment() 139 DeviceAssignment assignment(num_replicas, num_partitions); in GetDefaultDeviceAssignment() 141 for (int partition = 0; partition < num_partitions; ++partition) { in GetDefaultDeviceAssignment() 150 return placer.AssignDevices(num_replicas, num_partitions); in GetDefaultDeviceAssignment() 519 const int num_partitions = device_assignment_.computation_count(); in PyTpuExecutable() local 520 CHECK_EQ(num_partitions, 1) << "partition count > 1 is not supported."; in PyTpuExecutable() 522 for (int partition = 0; partition < num_partitions; ++partition) { in PyTpuExecutable() 628 if (num_partitions() != 1) { in Execute() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_module_config.h | 131 void set_num_partitions(int64 num_partitions) { in set_num_partitions() argument 132 num_partitions_ = num_partitions; in set_num_partitions() 134 int64 num_partitions() const { return num_partitions_; } in num_partitions() function
|
/external/tensorflow/tensorflow/lite/delegates/ |
D | utils_test.cc | 150 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in MockPreviewDelegatePartitioning() argument 153 *num_partitions = mock->num_delegate_params(); in MockPreviewDelegatePartitioning() 192 EXPECT_EQ(4, helper.num_partitions()); in TEST() 234 EXPECT_EQ(4, helper.num_partitions()); in TEST()
|