Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dynamic_partition_op_test.py163 num_partitions, argument
215 def testRuntimeError(self, data, partitions, num_partitions, error): argument
241 def testStaticError(self, data, partitions, num_partitions, error): argument
Dragged_array_ops.py547 def stack_dynamic_partitions(data, partitions, num_partitions, name=None): argument
/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 GetAndValidateAttributes()
124 absl::Span<const int32> num_partitions, in GetSliceIndices()
132 absl::Span<const int32> num_partitions, in GetSliceIndices()
141 absl::Span<const int32> num_partitions, in GetSliceIndices()
151 absl::Span<const int32> num_partitions, in GetSliceIndices()
164 absl::Span<const int32> num_partitions, in GetSliceIndices()
181 absl::Span<const int32> num_partitions, in GetSliceIndices()
202 absl::Span<const int32> num_partitions, in GetSliceIndices()
228 absl::Span<const int32> num_partitions, in GetSliceIndices()
259 absl::Span<const int32> num_partitions, in GetSliceIndices()
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()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc53 HloInstruction* hlo, int64_t num_partitions, const HloSharding& sharding, in PadEachPartitionWithHaloExchange()
113 int64_t num_partitions, in ShuffleWithinEachPartitionUsingOneHot()
160 HloInstruction* hlo, int64_t num_partitions, in ShuffleDataWithAllToAll()
172 HloInstruction* GetCorrectionFactor(HloInstruction* hlo, int64_t num_partitions, in GetCorrectionFactor()
230 int64_t num_partitions, HloInstruction* partition_id, in GetFinalFftUsingCollectivePermute()
Dconvolution_handler.cc48 int64_t num_partitions, SpmdBuilder* b) { in PartitionConvolutionWithBatchGroupCount()
140 int64_t num_partitions, SpmdBuilder* b) { in PartitionConvolutionWithFeatureGroupCount()
823 int64_t num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolutionBaseCase()
971 int64_t num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolution()
Ddot_handler.cc378 int64_t num_partitions, int64_t output_lhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()
514 int64_t num_partitions, in PartitionBaseCase()
1740 int64_t num_partitions, int64_t lhs_contracting_partitions, in PartitionDotGroupOnBatch()
2078 const DotConvDimsMapping& dims_mapping, int64_t num_partitions, in PartitionDotGroupOnNonContracting()
2266 const DotConvDimsMapping& dims_mapping, int64_t num_partitions, in PartitionDotGroupOnContracting()
2485 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2619 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in PrioritizeContractingDimensionsPartitioning()
2766 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()
2822 int64_t num_partitions, in PartitionDot()
3173 int64_t num_partitions, in PartitionDot()
Dspmd_partitioner.h197 SpmdPartitioner(int64_t num_partitions, int64_t num_replicas, in SpmdPartitioner()
531 int64 num_partitions; member
/external/webrtc/modules/audio_processing/aec3/
Dadaptive_fir_filter.cc36 size_t num_partitions, in ComputeFrequencyResponse()
60 size_t num_partitions, in ComputeFrequencyResponse_Neon()
92 size_t num_partitions, in ComputeFrequencyResponse_Sse2()
126 size_t num_partitions, in AdaptPartitions()
149 size_t num_partitions, in AdaptPartitions_Neon()
215 size_t num_partitions, in AdaptPartitions_Sse2()
282 size_t num_partitions, in ApplyFilter()
309 size_t num_partitions, in ApplyFilter_Neon()
379 size_t num_partitions, in ApplyFilter_Sse2()
715 void AdaptiveFirFilter::SetFilter(size_t num_partitions, in SetFilter()
/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/client/lib/
Dsorting.cc41 int64_t num_partitions = in TopK() local
80 XlaOp TopKWithPartitions(XlaOp input, int64_t k, int64_t num_partitions) { in TopKWithPartitions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fork_join.cc64 void** buffer_table, uint64* prof_counters, int32_t num_partitions, in __xla_cpu_runtime_ParallelForkJoin()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_spmd_partitioner_test.cc40 absl::string_view hlo_module, int64_t num_partitions, in PartitionComputation()
Dgpu_reduce_scatter_creator_test.cc41 int64_t num_partitions, in RunPass()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h132 void set_num_partitions(int64_t num_partitions) { in set_num_partitions()
135 int64 num_partitions() const { return num_partitions_; } in num_partitions() function
/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad.cc66 int32_t num_partitions; in DynamicPartitionGrad() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc69 int num_partitions) { in set_num_partitions()
Dexecutable_build_options.h84 int num_partitions() const { return num_partitions_; } in num_partitions() function
/external/tensorflow/tensorflow/lite/delegates/
Dutils.h96 int num_partitions() const { return partitions_.size(); } in num_partitions() function
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc327 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
384 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
447 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
561 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
728 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
948 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1017 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1214 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
/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/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc45 std::vector<std::vector<std::shared_ptr<PjRtDevice>>>> { in PYBIND11_MODULE()

123