Home
last modified time | relevance | path

Searched defs:outputIndex (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DEagerOperation.java70 public long getUnsafeNativeHandle(int outputIndex) { in getUnsafeNativeHandle()
75 public long[] shape(int outputIndex) { in shape()
91 public DataType dtype(int outputIndex) { in dtype()
103 public Tensor<?> tensor(int outputIndex) { in tensor()
117 private Tensor<?> resolveTensor(int outputIndex) { in resolveTensor()
/external/armnn/src/backends/reference/workloads/
DPad.cpp98 auto outputIndex = w + std::get<0>(padList[0]); in Pad() local
115 … auto outputIndex = (h + std::get<0>(padList[0])) * outputWidth + (w + std::get<0>(padList[1])); in Pad() local
137 … auto outputIndex = (c + std::get<0>(padList[0])) * outputHeight * outputWidth in Pad() local
169 auto outputIndex = (b + std::get<0>(padList[0])) in Pad() local
DPooling3d.cpp144 int outputIndex = n * depth * height * width * channels + in CalculateIndex() local
153 int outputIndex = n * channels * depth * height * width + in CalculateIndex() local
281 … int outputIndex = CalculateIndex(channels, depthOutput, heightOutput, widthOutput, in Pooling3d() local
316 … int outputIndex = CalculateIndex(channels, depthOutput, heightOutput, widthOutput, in Pooling3d() local
DTransposeConvolution2d.cpp81 unsigned int outputIndex; in TransposeConvolution2dImpl() local
145 const unsigned int outputIndex = in TransposeConvolution2dImpl() local
DPooling2d.cpp221 int outputIndex; in Pooling2d() local
280 int outputIndex; in Pooling2d() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
Dlhlo_elemental_utils.cc118 ValueRange outputIndex, in elementalLower()
155 ValueRange outputIndex, in elementalLowerImplForBroadcastInDimOps()
208 ValueRange outputIndex, bool checkCache) { in elementalLower()
216 ValueRange outputIndex, in elementalLower()
/external/armnn/delegate/classic/src/
DUnpack.hpp126 for (unsigned int outputIndex = 0; outputIndex < outputTensorInfos.size(); ++outputIndex) in VisitUnpackOperator() local
190 …for (unsigned int outputIndex = 0; outputIndex < splitterLayer->GetNumOutputSlots(); ++outputIndex) in VisitUnpackOperator() local
DClassicDelegateUtils.hpp170 for (unsigned int outputIndex = 0; outputIndex < layer->GetNumOutputSlots(); ++outputIndex) in Connect() local
260 …for (unsigned int outputIndex = 0; outputIndex < activationLayer->GetNumOutputSlots(); ++outputInd… in FusedActivation() local
306 … for (unsigned int outputIndex = 0; outputIndex < reshapeLayer->GetNumOutputSlots(); ++outputIndex) in AddReshapeLayer() local
DSplit.hpp146 for (unsigned int outputIndex = 0; outputIndex < layer->GetNumOutputSlots(); ++outputIndex) in VisitSplitOperator() local
/external/icu/icu4j/tools/misc/src/com/ibm/icu/dev/tool/layout/
DOpenTypeTableWriter.java27 OpenTypeTableDumper(short[] data, int outputIndex) in OpenTypeTableDumper()
72 protected int outputIndex; field in OpenTypeTableWriter
/external/armnn/tests/ExecuteNetwork/
DTfliteExecutor.cpp133 …for (unsigned int outputIndex = 0; outputIndex < m_TfLiteInterpreter->outputs().size(); ++outputIn… in Execute() local
233 …for (unsigned int outputIndex = 0; outputIndex < m_TfLiteInterpreter->outputs().size(); ++outputIn… in CompareAndPrintResult() local
/external/armnn/tests/
DInferenceModel.hpp563 unsigned int GetOutputSize(unsigned int outputIndex = 0u) const in GetOutputSize()
582 unsigned int outputIndex = i; in Run() local
650 unsigned int outputIndex = i; in RunAsync() local
702 unsigned int outputIndex = i; in RunAsync() local
738 const armnn::BindingPointInfo& GetOutputBindingInfo(unsigned int outputIndex = 0u) const in GetOutputBindingInfo()
749 QuantizationParams GetQuantizationParams(unsigned int outputIndex = 0u) const in GetQuantizationParams()
/external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/model/
DModel.java236 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor()
245 public int[] getOutputTensorShape(int outputIndex) { in getOutputTensorShape()
/external/armnn/delegate/opaque/src/
DOpaqueDelegateUtils.hpp217 for (unsigned int outputIndex = 0; outputIndex < layer->GetNumOutputSlots(); ++outputIndex) in Connect() local
312 …for (unsigned int outputIndex = 0; outputIndex < activationLayer->GetNumOutputSlots(); ++outputInd… in FusedActivation() local
377 … for (unsigned int outputIndex = 0; outputIndex < reshapeLayer->GetNumOutputSlots(); ++outputIndex) in AddReshapeLayer() local
/external/armnn/src/armnn/layers/
DMeanLayer.cpp98 unsigned int outputIndex = 0; in InferOutputShapes() local
DReduceLayer.cpp99 unsigned int outputIndex = 0; in InferOutputShapes() local
/external/oboe/samples/RhythmGame/src/main/cpp/audio/
DNDKExtractor.cpp149 ssize_t outputIndex = AMediaCodec_dequeueOutputBuffer(codec, &info, 0); in decode() local
/external/armnn/shim/sl/canonical/
DModelToINetworkTransformer.cpp177 uint32_t outputIndex = m_Model.main.outputIndexes[i]; in Convert() local
/external/android-nn-driver/
DModelToINetworkConverter.cpp192 uint32_t outputIndex = getMainModel(m_Model).outputIndexes[i]; in Convert() local
/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DInterpreterImpl.java160 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor()
DNativeSignatureRunnerWrapper.java68 int outputIndex = nativeGetOutputIndex(signatureRunnerHandle, outputName); in getOutputIndex() local
DInterpreterApi.java493 Tensor getOutputTensor(int outputIndex); in getOutputTensor()
/external/angle/src/libANGLE/renderer/d3d/
DDynamicHLSL.h83 size_t outputIndex = 0; member
/external/angle/src/libANGLE/
DProgramExecutable.cpp423 for (size_t outputIndex = 0; outputIndex < outputCount; ++outputIndex) in load() local
434 for (size_t outputIndex = 0; outputIndex < outputVarCount; ++outputIndex) in load() local
447 for (size_t outputIndex = 0; outputIndex < outputTypeCount; ++outputIndex) in load() local
462 for (size_t outputIndex = 0; outputIndex < secondaryOutputVarCount; ++outputIndex) in load() local
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorConvolution.h141 Index outputIndex = 0; in mapGpuOutputPlaneToTensorOutputOffset() local

12