Searched refs:pad_v2 (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | node_file_writer_test.py | 178 gen_array_ops.pad_v2(x, paddings, constant_values)
211 gen_array_ops.pad_v2(x, paddings, constant_values)
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | meta_spmd_expander.cc | 298 if (auto pad_v2 = llvm::dyn_cast<mlir::TF::PadV2Op>(op)) { in ComputeLayoutForward() local
299 pad_output = pad_v2.output(); in ComputeLayoutForward()
300 pad_input = pad_v2.input(); in ComputeLayoutForward()
322 if (auto pad_v2 = llvm::dyn_cast<mlir::TF::PadV2Op>(op)) { in ComputeLayoutBackward() local
323 pad_output = pad_v2.output(); in ComputeLayoutBackward()
324 pad_input = pad_v2.input(); in ComputeLayoutBackward()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 275 return array_ops.pad_v2(x1, padding, mode="CONSTANT")
|
| D | pfor.py | 2434 return wrap(array_ops.pad_v2(t, paddings, mode="CONSTANT"), True)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3514 def pad_v2(tensor, paddings, mode="CONSTANT", constant_values=0, name=None): function
3639 result = gen_array_ops.pad_v2(
|