Searched refs:new_spatial_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 1928 std::vector<int64_t> new_spatial_dims = retval.second; in Propagate() local
1934 new_spatial_dims, old_batch_dim, old_spatial_dims)); in Propagate()
2011 std::vector<int64_t> new_spatial_dims = retval.second; in Propagate() local
2018 const int64_t new_space_dim = new_spatial_dims[0]; in Propagate()
2037 new_batch_dim, new_spatial_dims, old_batch_dim, in Propagate()
2053 new_spatial_dims, new_space_size - extra_space)); in Propagate()
2058 first_operand, new_batch_dim, old_batch_size, new_spatial_dims, in Propagate()
2073 HaloDuplicateWithSlice(first_operand, new_spatial_dims, new_batch_dim, in Propagate()
2547 std::vector<int64_t> new_spatial_dims = old_new_dims.second; in PropagateOnConv() local
2581 activations_batch_dim, &new_spatial_dims)); in PropagateOnConv()
[all …]
|