Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h148 int num_partitions_ = 1; variable
/external/tensorflow/tensorflow/core/kernels/
Ddynamic_partition_op.cc82 int num_partitions_; member in tensorflow::DynamicPartitionOp_Shared
Ddynamic_partition_op_gpu.cu.cc472 int32 num_partitions_; member in tensorflow::DynamicPartitionOpGPU
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h246 const int64 num_partitions_; variable
524 int64 num_partitions_; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h257 int64 num_partitions_ = 1; variable
/external/tensorflow/tensorflow/lite/delegates/
Dutils.cc56 int num_partitions_ = 0; in Partition() local