Home
last modified time | relevance | path

Searched defs:output_desc (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbroadcast_to.h25 const NdArrayDesc<N>& output_desc, char* output_data, in BroadcastImpl()
67 NdArrayDesc<N> output_desc; in BroadcastTo() local
Dsub.h82 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
122 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
161 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
198 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
235 NdArrayDesc<N> output_desc; in BroadcastSub16POTSlow() local
281 NdArrayDesc<N> output_desc; in BroadcastQuantSubSlow() local
Dmaximum_minimum.h46 NdArrayDesc<N> output_desc; in MaximumMinimumBroadcastSlow() local
Dtranspose.h39 NdArrayDesc<N> output_desc; in TransposeImpl() local
Ddiv.h114 NdArrayDesc<N> output_desc; in BroadcastDivSlowQuantized() local
202 NdArrayDesc<N> output_desc; in BroadcastDivSlow() local
/external/ComputeLibrary/src/graph/nodes/
DFlattenLayerNode.cpp59 TensorDescriptor output_desc = src->desc(); in configure_output() local
DDequantizationLayerNode.cpp61 TensorDescriptor output_desc = src->desc(); in configure_output() local
DDummyNode.cpp62 TensorDescriptor output_desc = src->desc(); in configure_output() local
DReshapeLayer.cpp60 TensorDescriptor output_desc = src->desc(); in configure_output() local
DBoundingBoxTransformLayerNode.cpp67 TensorDescriptor output_desc = deltas->desc(); in configure_output() local
DGenerateProposalsLayerNode.cpp69 TensorDescriptor output_desc = src->desc(); in configure_output() local
DPermuteLayerNode.cpp67 TensorDescriptor output_desc = src->desc(); in configure_output() local
DDetectionPostProcessLayerNode.cpp68 TensorDescriptor output_desc; in configure_output() local
DPrintLayerNode.cpp77 TensorDescriptor output_desc = src->desc(); in configure_output() local
DPadLayerNode.cpp72 TensorDescriptor output_desc = src->desc(); in configure_output() local
DROIAlignLayerNode.cpp70 TensorDescriptor output_desc = src->desc(); in configure_output() local
DL2NormalizeLayerNode.cpp70 TensorDescriptor output_desc = src->desc(); in configure_output() local
DSliceLayerNode.cpp57 TensorDescriptor output_desc = input_descriptor; in compute_output_descriptor() local
DResizeLayerNode.cpp72 TensorDescriptor output_desc = src->desc(); in configure_output() local
DDepthToSpaceLayerNode.cpp81 TensorDescriptor output_desc = src->desc(); in configure_output() local
DStridedSliceLayerNode.cpp73 TensorDescriptor output_desc = input_descriptor; in compute_output_descriptor() local
/external/pytorch/torch/csrc/jit/codegen/fuser/cuda/
Dfused_kernel.cpp89 std::vector<TensorDesc> output_desc, in FusedKernelCUDA()
250 std::vector<TensorDesc> output_desc, in createFusionKernel()
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc144 const se::dnn::BatchDescriptor& output_desc, in LogConvAutotuneResults()
187 const se::dnn::BatchDescriptor& output_desc, in LogFusedConvForwardAutotuneResults()
Dconv_ops_gpu.cc113 const se::dnn::BatchDescriptor& output_desc, in AutotuneFusedConv()
280 const se::dnn::BatchDescriptor& output_desc, se::DeviceMemory<T> output_ptr, in AutotuneUnfusedConv()
/external/pytorch/torch/csrc/jit/codegen/fuser/cpu/
Dfused_kernel.cpp310 std::vector<TensorDesc> output_desc, in FusedKernelCPU()
345 std::vector<TensorDesc> output_desc, in createFusionKernel()

12