Home
last modified time | relevance | path

Searched refs:setInputTensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQuantizedLSTMTest.cpp97 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputTensor, input_), in invoke()
99 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
102 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToForgetWeightsTensor, in invoke()
105 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToCellWeightsTensor, in invoke()
108 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToOutputWeightsTensor, in invoke()
111 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
114 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToForgetWeightsTensor, in invoke()
117 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToCellWeightsTensor, in invoke()
120 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToOutputWeightsTensor, in invoke()
124 setInputTensor(&execution, QuantizedLSTMCell::kInputGateBiasTensor, inputGateBias_), in invoke()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/coreml/objc/
H A DPTMCoreMLExecutor.mm28 [_inputProvider setInputTensor:ival.toTensor() forFeatureName:_featureNames[input_count]];
32 [_inputProvider setInputTensor:val.toTensor() forFeatureName:_featureNames[input_count]];
H A DPTMCoreMLFeatureProvider.h12 - (void)setInputTensor:(const at::Tensor&)tensor forFeatureName:(NSString*)name;
H A DPTMCoreMLFeatureProvider.mm21 - (void)setInputTensor:(const at::Tensor&)tensor forFeatureName:(NSString *)name {
/aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-java-proto-lite/android_common_apex33/javac/
Dfederated-compute-java-proto-lite.jarMETA-INF/ META-INF/MANIFEST.MF com/ com/google/ com/ ...
/aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-java-proto-lite/android_common_apex33/turbine/
Dfederated-compute-java-proto-lite.jarcom/google/fcp/client/http/JniHttpHeader$Builder.class <Unknown> package ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/combined/
DFederatedCompute.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx/ META- ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/jarjar/
DFederatedCompute.jarxla/Protocol.class Protocol.java package xla public final xla.Protocol extends java.lang ...
/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- ...