Searched refs:computation_shape (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 180 computation_shape: Optional[np.ndarray] = None,
183 return device_assignment(topology, computation_shape, computation_stride,
339 computation_shape: Optional[np.ndarray] = None,
394 if computation_shape is None:
395 computation_shape = np.array([1] * topology_rank, dtype=np.int32)
397 computation_shape = np.asarray(computation_shape, dtype=np.int32)
404 if computation_shape.shape != (topology_rank,):
406 topology_rank, computation_shape.shape))
411 if any(computation_shape < 1):
414 computation_shape))
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TPUReplicateMetadata.pbtxt | 38 name: "computation_shape"
99 name: "computation_shape"
167 name: "computation_shape"
242 name: "computation_shape"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TPUReplicateMetadata.pbtxt | 38 name: "computation_shape"
99 name: "computation_shape"
167 name: "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 | 4817 …_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 | 4817 …_per_replica\', \'topology\', \'use_tpu\', \'device_assignment\', \'computation_shape\', \'host_co…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | graph_pruning_preserve_ops.mlir | 12 …1) wraps "tf.TPUReplicateMetadata"() {allow_soft_placement = false, computation_shape = [], device…
|
| D | tpu_space_to_depth_pass.mlir | 24 …ate = "while/cluster_while_body_271", allow_soft_placement = false, computation_shape = [], device…
106 …_tpu_replicate = "cluster_eval_step", allow_soft_placement = false, computation_shape = [], device…
180 …_tpu_replicate = "cluster_eval_step", allow_soft_placement = false, computation_shape = [], device…
|
| D | tpu_cluster_formation.mlir | 394 … {_tpu_replicate = "cluster_test_fn", allow_soft_placement = false, computation_shape = [], device…
419 … {_tpu_replicate = "cluster_test_fn", allow_soft_placement = false, computation_shape = [], device…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/ |
| D | end_to_end.mlir | 12 …adata"() {_tpu_replicate = "cluster", allow_soft_placement = false, computation_shape = [], device…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-function-defs.pbtxt | 94 key: "computation_shape"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 245 const auto& computation_shape = in Evaluate() local
248 if (!Shape::Equal().MinorToMajorOnlyInLayout()(computation_shape, in Evaluate()
253 i, ShapeUtil::HumanStringWithLayout(computation_shape), in Evaluate()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 837 DefaultValuedAttr<I64ArrayAttr, "{}">:$computation_shape,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 81475 name: "computation_shape"
81536 name: "computation_shape"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 54001 name: "computation_shape"
|