Home
last modified time | relevance | path

Searched defs:inputTensor (Results 26 – 50 of 65) sorted by relevance

123

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupUCC.cpp959 at::Tensor& inputTensor, in _allgather_base()
1104 at::Tensor& inputTensor, in alltoall_base()
H A DProcessGroupWrapper.cpp494 at::Tensor& inputTensor, in alltoall_base()
H A DProcessGroupMPI.cpp698 at::Tensor& inputTensor, in alltoall_base()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DReduceSumTestImpl.cpp32 auto inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ReduceTestCommon() local
H A DReduceProdTestImpl.cpp31 auto inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ReduceTestCommon() local
H A DReductionTestImpl.cpp33 auto inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ReductionTestCommon() local
H A DArgMinMaxTestImpl.cpp30 std::vector<T> inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ArgMinMaxTestCommon() local
H A DBatchNormalizationTestImpl.cpp61 auto inputTensor = QuantizedVector<T>(inputValues, qScale, qOffset); in BatchNormTestImpl() local
H A DL2NormalizationTestImpl.cpp53 auto inputTensor = armnnUtils::QuantizedVector<T>(inputData, in L2NormalizationTestImpl() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDefaultAsyncExecuteTest.cpp111 std::unique_ptr<Workload> CreateWorkload(TensorInfo info, ITensorHandle* inputTensor, ITensorHandle… in CreateWorkload()
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Dkernel.cpp155 auto inputTensor = inputs[spec.get_input_tensor_index()]; in prepareRunArgs() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefUnidirectionalSequenceLstmWorkload.cpp68 auto inputTensor = reinterpret_cast<float*>(inputs[0]->Map()); in Execute() local
/aosp_15_r20/external/armnn/delegate/classic/src/
H A Darmnn_delegate.cpp492 const armnn::ConstTensor inputTensor(inputTensorInfo, tensor->data.data); in Invoke() local
/aosp_15_r20/external/armnn/delegate/opaque/src/
H A Darmnn_delegate.cpp574 const armnn::ConstTensor inputTensor(inputTensorInfo, TfLiteOpaqueTensorData(tensor)); in Invoke() local
/aosp_15_r20/external/armnn/tests/TfLiteYoloV3Big-Armnn/
H A DTfLiteYoloV3Big-Armnn.cpp67 armnn::ConstTensor inputTensor(inputBinding.second, inputData.data()); in MakeInputTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
H A DNativeInterpreterWrapper.java375 TensorImpl inputTensor = inputTensors[index]; in getInputTensor() local
/aosp_15_r20/external/pytorch/android/pytorch_android/src/androidTest/java/org/pytorch/
H A DPytorchTestBase.java88 final Tensor inputTensor = Tensor.fromBlob(inputTensorData, inputTensorShape); in testEqTensor() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/
DTestGpuNnapi.cpp842 uint32_t inputTensor = mModel.addOperand(&tensorType); in initialize() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
H A DInterpreterTest.java784 Tensor inputTensor = interpreter.getInputTensorFromSignature("x", "mul_add"); in testModelWithSignatureDef() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/java/tensorflowlite_java/android_common_apex30/javac/
Dtensorflowlite_java.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/tensorflow/ org/ ...
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/java/tensorflowlite_java/android_common_apex33/javac/
Dtensorflowlite_java.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/tensorflow/ org/ ...
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp4040 auto& inputTensor = inputs[0]; in ParseSplitV() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc4141 auto inputTensor = input.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/OnDevicePersonalization/android_common_apex33/withres/
DOnDevicePersonalization.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx.annotation_annotation- ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/AdServices/adservices/service-core/adservices-service-core/android_common_apex33/jarjar/
Dadservices-service-core.jarprivate_join_and_compute/util/proto/test/Test.class Test.java package private_join_and_compute.util. ...

123