Searched refs:PadInDim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 144 xla::XlaOp padded_lhs_pad = xla::PadInDim( in DoMirrorPadGrad()
152 xla::XlaOp padded_rhs_pad = xla::PadInDim( in DoMirrorPadGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 461 XlaOp PadInDim(XlaOp operand, XlaOp padding_value, int64 dimno, int64 pad_lo,
1070 friend XlaOp PadInDim(XlaOp operand, XlaOp padding_value, int64 dimno,
1680 XlaOp PadInDim(XlaOp operand, XlaOp padding_value, int64 dimno, int64 pad_lo,
|
| D | xla_builder.cc | 1093 XlaOp XlaBuilder::PadInDim(XlaOp operand, XlaOp padding_value, int64 dimno, in PadInDim() function in xla::XlaBuilder
4118 XlaOp PadInDim(XlaOp operand, XlaOp padding_value, int64 dimno, int64 pad_lo, in PadInDim() function
4120 return operand.builder()->PadInDim(operand, padding_value, dimno, pad_lo, in PadInDim()
|