Home
last modified time | relevance | path

Searched defs:num_partitions_ (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.h285 const int64_t num_partitions_; variable
599 int64_t num_partitions_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h182 int num_partitions_ = 1; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddynamic_partition_op.cc82 int num_partitions_; member in tensorflow::DynamicPartitionOp_Shared
H A Ddynamic_partition_op_gpu.cu.cc462 int32 num_partitions_; member in tensorflow::DynamicPartitionOpGPU
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddynamic_partition_op.cc187 int64_t num_partitions_; member in tensorflow::__anonf96503a80111::DynamicPartitionOp
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dutils.cc59 int num_partitions_ = 0; in Partition() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.h563 int num_partitions_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h364 int64_t num_partitions_ = 1; variable