Home
last modified time | relevance | path

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

123

/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.py548 def stack_dynamic_partitions(data, partitions, num_partitions, name=None): argument
585 with ops.name_scope(name, 'SegmentStack', [data, partitions, num_partitions]):
593 num_partitions = ops.convert_to_tensor(
594 num_partitions, name='num_partitions', preferred_dtype=partitions.dtype)
597 num_partitions = math_ops.cast(num_partitions, partitions.dtype)
603 num_partitions.shape.assert_has_rank(0)
612 nrows=num_partitions,
624 num_partitions,
628 values, value_rowids, nrows=num_partitions, validate=False)
646 num_partitions)
/external/tensorflow/tensorflow/python/kernel_tests/
Ddynamic_partition_op_test.py44 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/
Dunpartition_embedding_lookup.cc60 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/
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()
Dir_function.cc271 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()
/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/compiler/xla/python/tpu_driver/client/
Dtpu_client.py95 options.num_partitions = compile_options.num_partitions
108 def get_default_device_assignment(self, num_replicas, num_partitions=None): argument
109 if num_partitions is not None:
111 num_partitions)
Dtpu_client.cc135 int num_replicas, int num_partitions) const { in GetDefaultDeviceAssignment()
136 if (num_partitions > 1) { in GetDefaultDeviceAssignment()
139 if (num_replicas * num_partitions <= local_device_count()) { in GetDefaultDeviceAssignment()
140 DeviceAssignment assignment(num_replicas, num_partitions); in GetDefaultDeviceAssignment()
142 for (int partition = 0; partition < num_partitions; ++partition) { in GetDefaultDeviceAssignment()
151 return placer.AssignDevices(num_replicas, num_partitions); in GetDefaultDeviceAssignment()
508 const int num_partitions = device_assignment_.computation_count(); in PyTpuExecutable() local
509 CHECK_EQ(num_partitions, 1) << "partition count > 1 is not supported."; in PyTpuExecutable()
511 for (int partition = 0; partition < num_partitions; ++partition) { in PyTpuExecutable()
616 if (num_partitions() != 1) { in Execute()
[all …]
Dtpu_client_extension.cc36 [](PyLocalClient* client, int num_replicas, int num_partitions) in PYBIND11_MODULE()
40 num_replicas, num_partitions)); in PYBIND11_MODULE()
44 result[r].resize(num_partitions); in PYBIND11_MODULE()
45 for (int p = 0; p < num_partitions; ++p) { in PYBIND11_MODULE()
/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/webrtc/webrtc/modules/audio_processing/aec/
Daec_core_mips.c326 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()
Daec_core_internal.h153 int num_partitions; member
177 int num_partitions,
190 int num_partitions,
Daec_core.c157 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 …]
/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/service/
Dhlo_module_config.h116 void set_num_partitions(int64 num_partitions) { in set_num_partitions() argument
117 num_partitions_ = num_partitions; in set_num_partitions()
119 int64 num_partitions() const { return num_partitions_; } in num_partitions() function
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDynamicPartition.pbtxt14 number_attr: "num_partitions"
17 name: "num_partitions"
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h76 int num_partitions() const { return num_partitions_; } in num_partitions() function
77 ExecutableBuildOptions& set_num_partitions(int num_partitions);
Dexecutable_build_options.cc68 int num_partitions) { in set_num_partitions() argument
69 num_partitions_ = num_partitions; in set_num_partitions()
/external/webrtc/webrtc/modules/rtp_rtcp/source/
Drtp_format_vp8_test_helper.cc39 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()
Dvp8_partition_aggregator.cc23 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()
Dvp8_partition_aggregator.h28 size_t num_partitions,
33 size_t num_partitions);
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_client.cc272 int num_replicas, int num_partitions) const { in GetDefaultDeviceAssignment()
274 num_partitions); in GetDefaultDeviceAssignment()
688 const int num_partitions = device_assignment_->computation_count(); in PyLocalExecutable() local
692 CHECK_EQ(num_partitions, executables_.size()) in PyLocalExecutable()
694 << " did not match number of partitions " << num_partitions; in PyLocalExecutable()
698 for (int partition = 0; partition < num_partitions; ++partition) { in PyLocalExecutable()
837 if (num_partitions() != 1) { in Execute()
840 num_partitions()); in Execute()
851 if (num_partitions() != 1) { in ExecutePerReplica()
855 num_partitions()); in ExecutePerReplica()
[all …]
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_grad.py35 num_partitions = op.get_attr("num_partitions")
41 original_indices, indices, num_partitions)

123