Searched refs:MakeEdgePaddingConfig (Results 1 – 6 of 6) sorted by relevance
129 xla::MakeEdgePaddingConfig({{0, 1}})); in XlaScatter()148 xla::MakeEdgePaddingConfig({{0, buffer_shape_post_axes.size()}})); in XlaScatter()154 xla::MakeEdgePaddingConfig({{0, buffer_shape_post_axes.size()}})); in XlaScatter()
157 xla::MakeEdgePaddingConfig({{0, elem_shape.dims()}})); in Compile()212 xla::MakeEdgePaddingConfig({{0, stack_shape.dims() - 1}})); in Compile()
126 xla::MakeEdgePaddingConfig({{0, 1}})); in XlaGather()136 xla::MakeEdgePaddingConfig( in XlaGather()
224 xla::MakeEdgePaddingConfig({{0, elem_shape.dims()}})); in Compile()268 xla::MakeEdgePaddingConfig({{0, ta_shape.dims() - 1}})); in Compile()422 xla::MakeEdgePaddingConfig({{0, elem_shape.dims()}})); in Compile()
215 PaddingConfig MakeEdgePaddingConfig( in MakeEdgePaddingConfig() function
341 PaddingConfig MakeEdgePaddingConfig(