Home
last modified time | relevance | path

Searched refs:new_space_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc1422 const int64 new_space_size = new_shape.dimensions(new_space_dim); in Propagate() local
1426 const int64 last_overlap_point = ((new_space_size - 1) / stride) * stride; in Propagate()
1428 << window_size << " new_space_size " << new_space_size; in Propagate()
1430 const int64 halo_size = last_overlap_point + window_size - new_space_size; in Propagate()
1437 new_space_size, init_val)); in Propagate()
1504 start_indices[new_space_dim] = new_space_size; in Propagate()
1505 end_indices[new_space_dim] = new_space_size + halo_size; in Propagate()
1565 ->set_edge_padding_high(new_space_size); in Propagate()
1574 tensorflow::core::Bitmap b(batch_size * (new_space_size + halo_size)); in Propagate()
1575 for (int k = 0; k < batch_size * (new_space_size + halo_size); ++k) { in Propagate()
[all …]