Searched refs:padded_input (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.cc | 305 Tensor padded_input; in Compute() local
319 &padded_input)); in Compute()
344 To32Bit(padded_input.tensor<T, 4>()), data_format)); in Compute()
345 transformed_input = padded_input; in Compute()
595 Tensor padded_input; in Compute() local
618 &padded_input)); in Compute()
651 To32Bit(padded_input.tensor<T, 4>()), in Compute()
654 transformed_input = padded_input; in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | slice_op.cc | 160 auto padded_input = xla::Pad( in Compile() local
165 auto sliced = xla::DynamicSlice(padded_input, scalar_indices, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 256 HloInstruction* padded_input = in CanonicalizeBackwardFilterConvolution() local
263 backward_conv->shape(), {padded_input, output})); in CanonicalizeBackwardFilterConvolution()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1106 padded_input = control_flow_ops.cond(
1111 padded_input = array_ops.pad(input_tensor, paddings)
1114 padded_input.op._set_attr( # pylint: disable=protected-access
1118 padded_inputs[core_idx].append(padded_input)
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOps.td | 294 padded_input[b,
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 6617 auto padded_input = in TEST_F() local
6624 AllOf(op::Slice(op::AllToAll(op::Dot(padded_input, op::Convert()))), in TEST_F()
|