Searched refs:computation_shape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 170 computation_shape=None, argument
173 return device_assignment(topology, computation_shape, computation_stride,
178 computation_shape=None, argument
227 if computation_shape is None:
228 computation_shape = np.array([1] * topology_rank, dtype=np.int32)
230 computation_shape = np.asarray(computation_shape, dtype=np.int32)
237 if computation_shape.shape != (topology_rank,):
239 topology_rank, computation_shape.shape))
244 if any(computation_shape < 1):
247 computation_shape))
[all …]
|
| D | tpu_context.py | 290 computation_shape=self._computation_shape,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TPUReplicateMetadata.pbtxt | 35 name: "computation_shape"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 27 …argspec: "args=[\'topology\', \'computation_shape\', \'computation_stride\', \'num_replicas\'], va…
|
| D | tensorflow.raw_ops.pbtxt | 3853 …_per_replica\', \'topology\', \'use_tpu\', \'device_assignment\', \'computation_shape\', \'host_co…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 27 …argspec: "args=[\'topology\', \'computation_shape\', \'computation_stride\', \'num_replicas\'], va…
|
| D | tensorflow.raw_ops.pbtxt | 3853 …_per_replica\', \'topology\', \'use_tpu\', \'device_assignment\', \'computation_shape\', \'host_co…
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 243 const auto& computation_shape = in Evaluate() local
246 if (!ShapeUtil::Equal(computation_shape, arg_shape)) { in Evaluate()
250 i, ShapeUtil::HumanStringWithLayout(computation_shape), in Evaluate()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 82355 name: "computation_shape"
82416 name: "computation_shape"
|
| D | ops_history.v1.pbtxt | 82982 name: "computation_shape"
83043 name: "computation_shape"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 40032 name: "computation_shape"
|