| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | dynamic_shapes.py | 751 def check_shape(path, t, dynamic_shape): argument
907 def transform_shape(path, t, dynamic_shape): argument
1039 def assoc_shape(path, t, dynamic_shape): argument
|
| /aosp_15_r20/external/executorch/examples/llm_manual/ |
| H A D | export_nanogpt.py | 25 dynamic_shape = ({1: torch.export.Dim("token_dim", max=model.config.block_size)},) variable
|
| /aosp_15_r20/external/executorch/backends/xnnpack/utils/ |
| H A D | configs.py | 39 dynamic_shape=False, argument
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/models/ |
| H A D | w2l.py | 22 dynamic_shape = ({0: torch.export.Dim("batch", min=2, max=10)},) variable in TestW2L
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/ops/ |
| H A D | conv1d.py | 96 dynamic_shape=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_util.cc | 93 for (auto& dynamic_shape : dynamic_shapes) { in DynamicShapesToTensorShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | shape_simplification.cc | 192 ShapeOfOp dynamic_shape; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_resource_ops_test.cc | 61 bool dynamic_shape; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | executable.cc | 44 Status ExecutionInput::SetDynamicShape(Shape dynamic_shape) { in SetDynamicShape()
|
| H A D | shape_inference_test.cc | 2386 auto dynamic_shape = ShapeUtil::MakeShape(S32, {4}, {true}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_execute.cc | 169 bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape, in DynamicShapeIsCompatible()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_util.cc | 1596 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicArrayShapeIsCompatible()
1609 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicShapeIsCompatible()
|
| H A D | literal_test.cc | 765 auto dynamic_shape = ShapeUtil::MakeShape(F32, {2, 3}, {false, true}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | strided_slice_op.cc | 604 xla::XlaOp dynamic_shape = ctx->Input(0); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | py_buffer.cc | 158 Shape dynamic_shape; in xla_dynamic_shape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | c_api_decl.h | 286 XLA_Shape dynamic_shape; member
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_compiled_autograd.py | 1301 def test_mismatch_fake_tensor_mode(self, dynamic_shape=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 765 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 1773 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
|