Home
last modified time | relevance | path

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

12345

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A 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 …]
H A Dtpu_compile_op_support.cc62 int num_partitions, const DebugOptions* debug_options, const int* seed, in CreateModuleConfig()
141 int num_partitions, const DebugOptions* debug_options) { in CreateModuleConfig()
H A Dsharding_util_ops_test.cc1218 const TensorShape& input_shape, absl::Span<const int32> num_partitions, in CreateRoundtripTensorGraph()
1267 const TensorShape& input_shape, absl::Span<const int32> num_partitions, in CreateRoundtripResourceGraph()
1361 const std::vector<int32> num_partitions(rank, 2); in TEST_P() local
1382 const std::vector<int32> num_partitions(rank, 2); in TEST_P() local
1403 const std::vector<int32> num_partitions(rank, 2); in TEST_P() local
/aosp_15_r20/external/webrtc/modules/audio_processing/aec3/
H A 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()
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A 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()
H A 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()
H A Dconvolution_handler.cc48 int64_t num_partitions, SpmdBuilder* b) { in PartitionConvolutionWithBatchGroupCount()
140 int64_t num_partitions, SpmdBuilder* b) { in PartitionConvolutionWithFeatureGroupCount()
825 int64_t num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolutionBaseCase()
973 int64_t num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolution()
H A Ddot_handler.cc435 int64_t num_partitions, int64_t output_lhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()
760 int64_t num_partitions, in PartitionBaseCase()
2008 int64_t num_partitions, int64_t lhs_contracting_partitions, in PartitionDotGroupOnBatch()
2354 const DotConvDimsMapping& dims_mapping, int64_t num_partitions, in PartitionDotGroupOnNonContracting()
2583 const DotConvDimsMapping& dims_mapping, int64_t num_partitions, in PartitionDotGroupOnContracting()
2845 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2990 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in PrioritizeContractingDimensionsPartitioning()
3218 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()
3346 int64_t num_partitions, in PartitionDot()
3734 int64_t num_partitions, in PartitionDot()
H A Dspmd_partitioner.h205 SpmdPartitioner(int64_t num_partitions, int64_t num_replicas, in SpmdPartitioner()
583 int64_t num_partitions; member
H A Dstateful_rng_spmd_partitioner.cc50 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in CreateVisitor()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A 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
H A Dragged_array_ops.py575 def stack_dynamic_partitions(data, partitions, num_partitions, name=None): argument
878 num_partitions, argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A 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()
H A Dgpu_spmd_partitioner.cc50 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in CreateVisitor()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dsorting.cc41 int64_t num_partitions = in TopK() local
153 XlaOp TopKWithPartitions(XlaOp input, int64_t k, int64_t num_partitions) { in TopKWithPartitions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A 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
H A Dcompile_options.proto34 int64 num_partitions = 5; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreduce_scatter_utils.cc253 const HloAllReduceInstruction* ar, int64_t num_partitions, in MatchReduceScatter()
269 const HloAllReduceInstruction* ar, int64_t num_partitions, in MatchReduceScatter()
H A Dhlo_module_config.h140 void set_num_partitions(int64_t num_partitions) { in set_num_partitions()
143 int64_t num_partitions() const { return num_partitions_; } in num_partitions() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dsharding_util_ops.cc42 std::vector<int64_t>& num_partitions, in GetAndValidateAttributes()
97 std::vector<int64_t> GetSliceIndices(absl::Span<const int64> num_partitions, in GetSliceIndices()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dlmhlo_to_jitrt.cc1160 CollectivePermuteOp op, int replica_count, int num_partitions) { in GetNcclCollectiveConfig()
1169 int replica_count, int num_partitions, PatternRewriter& rewriter) { in TryDegenerateToMemCopy()
1183 int replica_count, int num_partitions, PatternRewriter& rewriter) { in TryDegenerateToMemCopy()
1282 const int64_t num_partitions = num_partitions_attr.getInt(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dhlo_module_loader.h32 int64_t num_partitions = 1; member

12345