Home
last modified time | relevance | path

Searched defs:dst_tensors (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
H A DGpuKernelComponentGroup.cpp75 const auto dst_tensors = component->tensors().get_const_dst_tensors(); in add_component() local
97 const auto dst_tensors = component->tensors().get_const_dst_tensors(); in add_component() local
144 const auto dst_tensors = tensors.get_const_dst_tensors(); in finalize() local
230 const auto dst_tensors = component->tensors().get_const_dst_tensors(); in finalize() local
H A DGpuOperatorGroup.cpp106 const auto dst_tensors = op.tensors().get_const_dst_tensors(); in try_add_operator() local
128 const auto dst_tensors = op.tensors().get_const_dst_tensors(); in try_add_operator() local
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
H A DDependencyGraph.h427 std::vector<TensorId> dst_tensors(OperatorId op) const in dst_tensors() function
605 const auto dst_tensors = _adj_dst_tensors.at(op); in build_operators_sequence_from_op() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dgpu_operation.h69 std::vector<TensorDescriptor> dst_tensors; member
H A Dserialization_base_generated.h1562 …const flatbuffers::Vector<flatbuffers::Offset<tflite::gpu::data::TensorDescriptor>> *dst_tensors()… in dst_tensors() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
H A Dperformance_profiling.cc178 std::vector<TensorFloat32> dst_tensors(graph_cl.outputs().size()); in RunSerializedTest() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_weights_converter_test_util.cc70 std::vector<TensorFloat32> dst_tensors; in ConvolutionWeightsConverterTest() local
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp113 std::vector<Tensor> dst_tensors; in _broadcast_out_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_model.cc755 std::vector<DummySpatialTensor> dst_tensors(node.outputs.size()); in ResolvePolymorphicArgs() local