Searched refs:rhs_pad (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 57 auto rhs_pad = xla::SliceInDim(t_rev, excluded_edges, in DoMirrorPad() local
59 accum = xla::ConcatInDim(b, {lhs_pad, accum, rhs_pad}, dimno); in DoMirrorPad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 3214 auto* rhs_pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P() local
3223 F32, {1024, rhs_pad->shape().dimensions(0), 100, 100}), // bf01 in TEST_P()
3229 rhs_pad->shape().dimensions(2), in TEST_P()
3230 rhs_pad->shape().dimensions(3), in TEST_P()
3241 ShapeInference::InferConvolveShape(input->shape(), rhs_pad->shape(), in TEST_P()
3245 input, rhs_pad, /*feature_group_count=*/1, /*batch_group_count=*/1, in TEST_P()
|