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