HomeSort by relevance Sort by last modified time
    Searched defs:num_cores_per_replica (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
device_assignment.py 153 def num_cores_per_replica(self): member in class:DeviceAssignment
  /external/tensorflow/tensorflow/compiler/tf2xla/
tf2xla_util_test.cc 228 const int num_cores_per_replica = 4; local
236 auto parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica);
244 parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica);
251 parse_status = ParseShardingFromDevice(*a_node, num_cores_per_replica);

Completed in 297 milliseconds