| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEPadLayer.cpp | 181 …const TensorShape padded_shape = misc::shape_calculator::compute_padded_shape(input->info()->tenso… in configure() local
217 …const TensorShape padded_shape = misc::shape_calculator::compute_padded_shape(input->tensor_shape(… in validate() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorTransformerFunctions.cpp | 268 std::vector<int64_t> padded_shape; in _jagged_to_padded_dense_forward_cpu() local
323 auto padded_shape = padded.sizes(); in _padded_dense_to_jagged_forward_cpu() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | padded_batch_dataset_op.cc | 289 const PartialTensorShape& padded_shape = in CopyBatch() local
451 PartialTensorShape padded_shape; in MakeDataset() local
|
| H A D | padded_batch_dataset_op_test.cc | 54 for (auto& padded_shape : padded_shapes_) { in GetInputTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | resolve_constant_slice.cc | 66 Shape padded_shape = input_array.shape(); in Slice() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | PadLayer.cpp | 52 …const TensorShape padded_shape = misc::shape_calculator::compute_padded_shape(orig_shape, paddings… in pad_layer() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuCopyKernel.cpp | 70 … const TensorShape padded_shape = misc::shape_calculator::compute_padded_shape(src_shape, padding); in validate_and_configure_window_with_padding() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | spacetobatch_op.cc | 55 std::vector<int64_t> padded_shape(input_shape.begin(), input_shape.end()); in SpaceToBatch() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLPadLayerKernel.cpp | 57 …TensorShape padded_shape = misc::shape_calculator::compute_padded_shape(input->tensor_shape(), pad… in validate_arguments() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | tree_reduction_rewriter.cc | 148 Shape padded_shape = in RewriteReduction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_fused_image_transform.cc | 710 TensorShape padded_shape; in Compute() local
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/ |
| H A D | ShapeCalculator.h | 1237 TensorShape padded_shape = input_shape; in compute_padded_shape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | dot_handler.cc | 1149 auto padded_shape = padded_slice_operand->shape(); in PartitionBaseCase() local
4016 Shape padded_shape = user_gte->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
|
| H A D | spmd_partitioner.cc | 683 Shape padded_shape = base_shape_; in ReshardAsWindowedInput() local
1539 auto padded_shape = hlo_->shape(); in ReshardWithAllToAll() local
|
| H A D | spmd_partitioner_util.cc | 529 auto padded_shape = result->shape(); in PadFromPartialReplicateShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 5186 def _is_padded_shape_compatible_with(padded_shape, input_component_shape): argument
5210 def _padded_shape_to_tensor(padded_shape, input_component_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | lower_tf.cc | 907 auto padded_shape = llvm::to_vector<4>(input_shape); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_padder.cc | 905 Shape padded_shape = input->shape(); in RewriteInputWithDynamicPadding() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_common.cc | 781 SmallVector<int64_t> padded_shape(input_rank); in convertSpaceToBatchNDOp() local
|