Searched refs:pad_literal (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 31 const xla::LiteralSlice& pad_literal, in DoMirrorPad() argument
43 int64 lhs_padding = pad_literal.Get<int64>({dimno, 0}); in DoMirrorPad()
44 int64 rhs_padding = pad_literal.Get<int64>({dimno, 1}); in DoMirrorPad()
88 xla::Literal pad_literal; in Compile() local
90 ctx->ConstantInputAsInt64Literal("paddings", &pad_literal)); in Compile()
97 DoMirrorPad(in0, in0_shape.ValueOrDie(), pad_literal, mode, b); in Compile()
|
| D | pad_op.cc | 54 xla::Literal pad_literal; in Compile() local
56 ctx->ConstantInputAsInt64Literal("paddings", &pad_literal)); in Compile()
61 int before = pad_literal.Get<int64>({i, 0}); in Compile()
62 int after = pad_literal.Get<int64>({i, 1}); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3192 auto& pad_literal = pad_value->literal(); in HandleReduceWindow() local
3194 if (pad_literal == reduce_init_literal) { in HandleReduceWindow()
3198 pad_literal.ConvertToShape(reduce_init_value->shape()); in HandleReduceWindow()
|