Home
last modified time | relevance | path

Searched refs:_XLA_OP_BY_OP_INPUTS_LIMIT (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py71 _XLA_OP_BY_OP_INPUTS_LIMIT = 200 variable
1185 if len(value.values) <= _XLA_OP_BY_OP_INPUTS_LIMIT:
1189 value_list[:_XLA_OP_BY_OP_INPUTS_LIMIT], axis=axis)
1190 for i in range(_XLA_OP_BY_OP_INPUTS_LIMIT, len(value_list),
1191 _XLA_OP_BY_OP_INPUTS_LIMIT - 1):
1193 [output] + value_list[i:i + _XLA_OP_BY_OP_INPUTS_LIMIT - 1],
1249 if len(value.values) <= _XLA_OP_BY_OP_INPUTS_LIMIT:
1253 for i in range(0, len(value_list), _XLA_OP_BY_OP_INPUTS_LIMIT):
1254 output += math_ops.add_n(value_list[i:i + _XLA_OP_BY_OP_INPUTS_LIMIT])