Searched refs:lhs_pad (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 56 auto lhs_pad = in DoMirrorPad() local
61 accum = xla::ConcatInDim(b, {lhs_pad, accum, rhs_pad}, dimno); in DoMirrorPad()
142 xla::XlaOp lhs_pad = xla::SliceInDim(grad, 0, lhs_padding, 1, dimno); in DoMirrorPadGrad() local
143 xla::XlaOp reverse_lhs_pad = xla::Rev(lhs_pad, {dimno}); in DoMirrorPadGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 3866 auto* lhs_pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P() local
3875 F32, {lhs_pad->shape().dimensions(1), 256, 3, 3}), // io01 in TEST_P()
3885 lhs_pad->shape(), filter->shape(), in TEST_P()
3890 lhs_pad, filter, /*feature_group_count=*/1, /*batch_group_count=*/1, in TEST_P()
|