Home
last modified time | relevance | path

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

12

/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/spmd/
Dfft_handler.cc53 HloInstruction* hlo, int64 num_partitions, const HloSharding& sharding, in PadEachPartitionWithHaloExchange()
113 int64 num_partitions, in ShuffleWithinEachPartitionUsingOneHot()
160 HloInstruction* hlo, int64 num_partitions, in ShuffleDataWithAllToAll()
172 HloInstruction* GetCorrectionFactor(HloInstruction* hlo, int64 num_partitions, in GetCorrectionFactor()
230 int64 num_partitions, HloInstruction* partition_id, int64* next_channel_id, in GetFinalFftUsingCollectivePermute()
Dconvolution_handler.cc48 int64 num_partitions, SpmdBuilder* b) { in PartitionConvolutionWithBatchGroupCount()
140 int64 num_partitions, SpmdBuilder* b) { in PartitionConvolutionWithFeatureGroupCount()
823 int64 num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolutionBaseCase()
971 int64 num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolution()
Ddot_handler.cc327 int64 FirstShardingDimWithPartitionOfSize(int64 num_partitions, in FirstShardingDimWithPartitionOfSize()
384 int64 num_partitions, int64 output_lhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()
469 int64 num_partitions, in PartitionBaseCase()
1651 int64 num_partitions, int64 lhs_contracting_partitions, in PartitionDotGroupOnBatch()
1987 const DotConvDimsMapping& dims_mapping, int64 num_partitions, in PartitionDotGroupOnNonContracting()
2082 int64 num_partitions, in PartitionDotGroupOnContracting()
2291 int64 num_partitions, int64 lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()
2434 int64 num_partitions, in PartitionDot()
2801 int64 num_partitions, in PartitionDot()
/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/python/keras/integration_test/
Dgradient_checkpoint_test.py29 def _get_big_cnn_model(img_dim, n_channels, num_partitions, argument
48 def _get_split_cnn_model(img_dim, n_channels, num_partitions, argument
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc316 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
375 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
438 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
552 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
719 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
939 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1011 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
1213 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
/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 num_partitions = in TopK() local
76 XlaOp TopKWithPartitions(XlaOp input, int64 k, int64 num_partitions) { in TopKWithPartitions()
/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()
Dir_function.cc278 const int64 num_partitions = partition_iterator.GetTotalPartitionCount(); in EmitCallToParallelForkJoin() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h131 void set_num_partitions(int64 num_partitions) { in set_num_partitions()
134 int64 num_partitions() const { return num_partitions_; } in num_partitions() function
Dhlo_replication_analysis.cc83 const int64 num_partitions = in DetermineHloInstructionIsReplicated() local
/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad.cc66 int32 num_partitions; in DynamicPartitionGrad() local
/external/tensorflow/tensorflow/lite/delegates/
Dutils.h92 int num_partitions() const { return partitions_.size(); } in num_partitions() function
Dutils_test.cc150 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in MockPreviewDelegatePartitioning()
/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/toco/graph_transformations/
Dunpartition_embedding_lookup.cc161 int num_partitions = indices_partition_op->num_partitions; in Run() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc60 int num_partitions, const DebugOptions* debug_options, const int* seed, in CreateModuleConfig()
139 int num_partitions, const DebugOptions* debug_options) { in CreateModuleConfig()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc117 int* num_replicas, int* num_partitions, in ParseDeviceAssignmentCompileOptions()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc40 std::vector<std::vector<std::shared_ptr<PjRtDevice>>>> { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/ops/
Ddata_flow_ops.cc55 int64 num_partitions; in __anonea01102b0202() local
92 int32 num_partitions; in DynamicStitchShapeFunction() local

12