Home
last modified time | relevance | path

Searched defs:num_tpu_devices_per_task (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/tpu/
H A Dtopology.proto37 int32 num_tpu_devices_per_task = 3; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtpu_rewrite_device_util_test.cc123 int num_tpu_devices_per_task) { in TopologyWithMeshShapeAndTasks()