Searched refs:GetInferredShapeSize (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ | ||
| D | distributed_tpu_rewrite_pass.cc | 424 sizes_[i] = GetInferredShapeSize(shapes[i], types[i]); in TensorDevicePlacer() |