Searched defs:num_cores_per_replica (Results 1 – 6 of 6) sorted by relevance
45 Status CoreOutOfRangeError(int core, int num_cores_per_replica) { in CoreOutOfRangeError()53 const string& device_name, int num_cores_per_replica, in ParseShardingFromDevice()81 const NodeDef& node_def, int num_cores_per_replica) { in ParseShardingFromDevice()89 const Node& node, int num_cores_per_replica) { in ParseShardingFromDevice()
235 const int num_cores_per_replica = 4; in TEST() local
35 int32 num_cores_per_replica = 2; field
119 num_cores_per_replica=None, argument
109 def num_cores_per_replica(self): member in DeviceAssignment
90 int num_cores_per_replica = in Compile() local