Home
last modified time | relevance | path

Searched defs:num_partitions (Results 1 – 25 of 76) sorted by relevance

1234

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dynamic_partition_op_test.py159 num_partitions, argument
211 def testRuntimeError(self, data, partitions, num_partitions, error): argument
237 def testStaticError(self, data, partitions, num_partitions, error): argument
Dragged_array_ops.py575 def stack_dynamic_partitions(data, partitions, num_partitions, name=None): argument
878 num_partitions, argument
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dstateful_rng_spmd_partitioner.h32 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in StatefulRngSpmdPartitioningVisitor()
45 StatefulRngSpmdPartitioner(int64_t num_partitions, int64_t num_replicas) in StatefulRngSpmdPartitioner()
Dfft_handler.cc51 HloInstruction* hlo, int64_t num_partitions, const HloSharding& sharding, in PadEachPartitionWithHaloExchange()
111 int64_t num_partitions, in ShuffleWithinEachPartitionUsingOneHot()
158 HloInstruction* hlo, int64_t num_partitions, in ShuffleDataWithAllToAll()
170 HloInstruction* GetCorrectionFactor(HloInstruction* hlo, int64_t num_partitions, in GetCorrectionFactor()
228 int64_t num_partitions, HloInstruction* partition_id, in GetFinalFftUsingCollectivePermute()
Dstateful_rng_spmd_partitioner.cc50 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in CreateVisitor()
Dstateful_rng_spmd_partitioner_test.cc40 absl::string_view hlo_module, int64_t num_partitions, in PartitionComputation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_spmd_partitioner.h31 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in GpuSpmdPartitioningVisitor()
44 GpuSpmdPartitioner(int64_t num_partitions, int64_t num_replicas) in GpuSpmdPartitioner()
Dgpu_spmd_partitioner.cc50 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in CreateVisitor()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc50 std::vector<int32>& num_partitions, in GetAndValidateAttributesHelper()
107 std::vector<int32>& num_partitions, in GetAndValidateAttributes()
179 absl::Span<const int32> num_partitions, in GetSliceIndices()
187 absl::Span<const int32> num_partitions, in GetSliceIndices()
196 absl::Span<const int32> num_partitions, in GetSliceIndices()
206 absl::Span<const int32> num_partitions, in GetSliceIndices()
219 absl::Span<const int32> num_partitions, in GetSliceIndices()
236 absl::Span<const int32> num_partitions, in GetSliceIndices()
257 absl::Span<const int32> num_partitions, in GetSliceIndices()
283 absl::Span<const int32> num_partitions, in GetSliceIndices()
[all …]
/external/webrtc/modules/audio_processing/aec3/
Dadaptive_fir_filter.cc36 size_t num_partitions, in ComputeFrequencyResponse()
60 size_t num_partitions, in ComputeFrequencyResponse_Neon()
94 size_t num_partitions, in ComputeFrequencyResponse_Sse2()
130 size_t num_partitions, in AdaptPartitions()
153 size_t num_partitions, in AdaptPartitions_Neon()
219 size_t num_partitions, in AdaptPartitions_Sse2()
286 size_t num_partitions, in ApplyFilter()
313 size_t num_partitions, in ApplyFilter_Neon()
383 size_t num_partitions, in ApplyFilter_Sse2()
729 void AdaptiveFirFilter::SetFilter(size_t num_partitions, in SetFilter()
Dadaptive_fir_filter_avx2.cc23 size_t num_partitions, in ComputeFrequencyResponse_Avx2()
56 size_t num_partitions, in AdaptPartitions_Avx2()
122 size_t num_partitions, in ApplyFilter_Avx2()
/external/tensorflow/tensorflow/core/kernels/
Ddynamic_partition_op_test.cc162 Node* DynamicPartitionNode(Graph* g, Node* in0, Node* in1, int num_partitions) { in DynamicPartitionNode()
173 static Graph* DynamicPartition(int num_partitions, int dim) { in DynamicPartition()
Dsmooth-hinge-loss.h33 double ComputeUpdatedDual(const int num_partitions, const double label, in ComputeUpdatedDual()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_module_loader.h32 int64_t num_partitions = 1; member
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dcompile_options.proto34 int64 num_partitions = 5; field
Dpjrt_c_api_client.h146 int num_replicas, int num_partitions) const override { in GetDefaultDeviceAssignment()
434 int num_partitions() const override { return wrapped()->num_partitions(); } in num_partitions() function
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc41 int64_t num_partitions = in TopK() local
153 XlaOp TopKWithPartitions(XlaOp input, int64_t k, int64_t num_partitions) { in TopKWithPartitions()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_spmd_partitioner_test.cc40 absl::string_view hlo_module, int64_t num_partitions, in PartitionComputation()
/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad.cc66 int32_t num_partitions; in DynamicPartitionGrad() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc331 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
388 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
451 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
565 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
732 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
950 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1019 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1216 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1289 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1346 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
/external/tensorflow/tensorflow/lite/delegates/
Dutils.h96 int num_partitions() const { return partitions_.size(); } in num_partitions() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fork_join.cc63 int32_t num_partitions, int64_t* partitions, int32_t num_partitioned_dims, in __xla_cpu_runtime_ParallelForkJoin()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunpartition_embedding_lookup.cc161 int num_partitions = indices_partition_op->num_partitions; in Run() local
/external/tensorflow/tensorflow/dtensor/mlir/
Ddtensor_allreduce_scatter_optimization.cc59 const int32 num_partitions = partitions.size(); in GetScatterGroupAssignment() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc69 int num_partitions) { in set_num_partitions()

1234