Home
last modified time | relevance | path

Searched refs:num_partitions (Results 1 – 25 of 88) sorted by relevance

1234

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dynamic_partition_op_test.py43 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 …]
Dragged_array_ops.py547 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/
Dadaptive_fir_filter_unittest.cc68 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 …]
Dadaptive_fir_filter.h31 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,
Dadaptive_fir_filter.cc36 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/
Ddynamic_partition_op_test.py46 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/
Dunpartition_embedding_lookup.cc60 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/
Dparameter_server_strategy_v2.py597 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/
Druntime_fork_join.cc64 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/
Dfft_handler.cc53 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 …]
Ddot_handler.cc327 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/
Dapi_def_DynamicPartition.pbtxt6 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/
Dgradient_checkpoint_test.py29 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/
Dtpu_client.cc83 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/
Dsorting.cc41 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/
Ddynamic_partition_op_test.cc162 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()
Dsmooth-hinge-loss.h33 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/
Ddata_flow_grad.cc66 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/
Dexecutable_build_options.cc69 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/
Dmodel_builder_test.cc316 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/
DDynamicPartition.pbtxt14 number_attr: "num_partitions"
17 name: "num_partitions"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDynamicPartition.pbtxt14 number_attr: "num_partitions"
17 name: "num_partitions"
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc134 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/
Dhlo_module_config.h131 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/
Dutils_test.cc150 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()

1234