Home
last modified time | relevance | path

Searched defs:num_partitions (Results 1 – 25 of 38) 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.py548 def stack_dynamic_partitions(data, partitions, num_partitions, name=None): argument
Dragged_dispatch.py445 def _ragged_dynamic_partition(data, partitions, num_partitions, name=None): argument
/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/
Dexecutable_build_options.h76 int num_partitions() const { return num_partitions_; } in num_partitions() function
Dexecutable_build_options.cc68 int num_partitions) { in set_num_partitions()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h116 void set_num_partitions(int64 num_partitions) { in set_num_partitions()
119 int64 num_partitions() const { return num_partitions_; } in num_partitions() function
/external/webrtc/webrtc/modules/rtp_rtcp/source/
Dvp8_partition_aggregator.cc23 size_t num_partitions, in PartitionTreeNode()
39 size_t num_partitions) { in CreateRootNode()
Drtp_format_vp8_test_helper.cc39 size_t num_partitions) { in Init()
/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.cc271 const int64 num_partitions = partition_iterator.GetTotalPartitionCount(); in EmitCallToParallelForkJoin() local
/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad.cc66 int32 num_partitions; in DynamicPartitionGrad() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunpartition_embedding_lookup.cc159 int num_partitions = indices_partition_op->num_partitions; in Run() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.py108 def get_default_device_assignment(self, num_replicas, num_partitions=None): argument
Dtpu_client_extension.cc37 -> StatusOr<std::vector<std::vector<std::shared_ptr<Device>>>> { in PYBIND11_MODULE()
Dtpu_client.h293 int num_partitions() const { return device_assignment_.computation_count(); } in num_partitions() function
/external/u-boot/arch/arm/mach-snapdragon/
Ddram.c36 u32 num_partitions; member
/external/webrtc/webrtc/modules/audio_processing/aec/
Daec_core_mips.c326 int num_partitions, in WebRtcAec_FilterFar_mips()
443 int num_partitions, in WebRtcAec_FilterAdaptation_mips()
Daec_core.c157 int num_partitions, in FilterFar()
215 int num_partitions, in FilterAdaptation()
924 int num_partitions, in EchoSubtraction()
Daec_core_neon.c38 int num_partitions, in FilterFarNEON()
190 int num_partitions, in FilterAdaptationNEON()
Daec_core_sse2.c33 int num_partitions, in FilterFarSSE2()
152 int num_partitions, in FilterAdaptationSSE2()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_client.h316 int num_partitions() const { in num_partitions() function
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc466 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning()
500 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning()
1205 int* num_partitions) { return ForbiddenContextFunction(context); }; in SwitchToKernelContext()
/external/tensorflow/tensorflow/core/ops/
Ddata_flow_ops.cc55 int64 num_partitions; in __anonb84a3fcf0202() local
92 int32 num_partitions; in DynamicStitchShapeFunction() local

12