Home
last modified time | relevance | path

Searched defs:get_shape (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dpack.py63 def get_shape(parameters): function
H A Dconcat.py68 def get_shape(parameters, delta): function
H A Droll.py95 def get_shape(param): function
/aosp_15_r20/external/executorch/backends/transforms/
H A Daddmm_mm_to_linear.py17 def get_shape(input_node: torch.fx.Node): function
/aosp_15_r20/external/armnn/python/pyarmnn/examples/common/
H A Dnetwork_executor_tflite.py91 def get_shape(self): member in TFLiteNetworkExecutor
H A Dnetwork_executor.py108 def get_shape(self): member in ArmnnNetworkExecutor
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
H A Droll.py99 def get_shape(param): function
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dtf_should_use_test.py90 def get_shape(h): function
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dcompiler_utils.py41 def get_shape( function
/aosp_15_r20/external/pytorch/test/functorch/
H A Dfunctorch_additional_op_db.py356 def get_shape(dim): function
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsparse_tensor.py150 def get_shape(self): member in SparseTensor
H A Dops.py609 def get_shape(self): member in Tensor
1265 def get_shape(self): member in _EagerTensorBase
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dkeras_tensor.py217 def get_shape(self): member in KerasTensor
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py371 def get_shape(cls, lines: List[List["Segment"]]) -> Tuple[int, int]: member in Segment
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py371 def get_shape(cls, lines: List[List["Segment"]]) -> Tuple[int, int]: member in Segment
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py371 def get_shape(cls, lines: List[List["Segment"]]) -> Tuple[int, int]: member in Segment
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py371 def get_shape(cls, lines: List[List["Segment"]]) -> Tuple[int, int]: member in Segment
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dvariables.py1170 def get_shape(self): member in Variable
3038 def get_shape(self): member in PartitionedVariable
/aosp_15_r20/external/pytorch/torch/utils/
H A Dflop_counter.py19 def get_shape(i): function
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dautocast_variable.py342 def get_shape(self): member in AutoCastVariable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dgather_scatter_handler.cc710 int64_t ShapeSizeInBytesSum(absl::Span<const T> operands, F&& get_shape) { in ShapeSizeInBytesSum()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor.py920 def get_shape(self): member in RaggedTensor
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dvalues.py695 def get_shape(self): member in DistributedVariable
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc1002 auto get_shape = [this](Operation *op, in matchAndRewrite() local