Searched refs:new_dot_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 227 const Shape& new_dot_shape) { in GenNewConvDNums() argument
303 output_spatial_dimensions.push_back(new_dot_shape.dimensions_size() - 1); in GenNewConvDNums()
938 auto new_dot_shape = original_sharded_dot_shape; in PartitionBaseCase() local
939 std::vector<int64> new_dims(new_dot_shape.dimensions().begin(), in PartitionBaseCase()
940 new_dot_shape.dimensions().end()); in PartitionBaseCase()
951 new_dot_shape = in PartitionBaseCase()
957 new_dot_shape, dot_lhs, dot_rhs, new_ddnums, in PartitionBaseCase()
980 new_dot_shape, dot_lhs, dot_rhs, in PartitionBaseCase()
990 indices_map.rhs_to_output_indices, new_dot_shape), in PartitionBaseCase()
|