Searched defs:num_slices_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/ | ||
| D | group_assignment.h | 83 int num_slices_; variable |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
| D | sharding_util_ops.cc | 250 int num_slices_ = 1; member in tensorflow::__anonc5e13fd90111::XlaSplitNDBaseOp |
| /external/tensorflow/tensorflow/core/tpu/kernels/ | ||
| D | sharding_util_ops.cc | 471 int num_slices_; member in tensorflow::__anon55ecaccf0111::XlaSplitNDShared |