Home
last modified time | relevance | path

Searched defs:num_shards_ (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/context/
H A Dcontainer.h156 uint32_t num_shards_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Drange_sampler.h238 int32 num_shards_; variable
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/src/core/lib/event_engine/posix_engine/
Dtimer.h176 const size_t num_shards_; variable
/aosp_15_r20/external/grpc-grpc/src/core/lib/event_engine/posix_engine/
H A Dtimer.h149 const size_t num_shards_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dtensor_bundle.h329 int num_shards_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dshard_dataset_op_test.cc64 int64_t num_shards_; member in tensorflow::data::__anone53a3e170111::ShardDatasetParams
H A Dshard_dataset_op.cc257 const int64_t num_shards_; member in tensorflow::data::ShardDatasetOp::Dataset