Searched refs:rhs_pad (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 59 auto rhs_pad = xla::SliceInDim(t_rev, excluded_edges, in DoMirrorPad() local
61 accum = xla::ConcatInDim(b, {lhs_pad, accum, rhs_pad}, dimno); in DoMirrorPad()
149 xla::XlaOp rhs_pad = in DoMirrorPadGrad() local
151 xla::XlaOp reverse_rhs_pad = xla::Rev(rhs_pad, {dimno}); in DoMirrorPadGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 3976 auto* rhs_pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P() local
3985 F32, {1024, rhs_pad->shape().dimensions(0), 100, 100}), // bf01 in TEST_P()
3991 rhs_pad->shape().dimensions(2), in TEST_P()
3992 rhs_pad->shape().dimensions(3), in TEST_P()
4004 input->shape(), rhs_pad->shape(), in TEST_P()
4009 input, rhs_pad, /*feature_group_count=*/1, /*batch_group_count=*/1, in TEST_P()
|