Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
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
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
/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
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
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.h69 std::vector<TensorDescriptor> dst_tensors; member
Dserialization_base_generated.h1562 …const flatbuffers::Vector<flatbuffers::Offset<tflite::gpu::data::TensorDescriptor>> *dst_tensors()… in dst_tensors() function
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
Dperformance_profiling.cc178 std::vector<TensorFloat32> dst_tensors(graph_cl.outputs().size()); in RunSerializedTest() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc70 std::vector<TensorFloat32> dst_tensors; in ConvolutionWeightsConverterTest() local
/external/pytorch/torch/csrc/cuda/
Dcomm.cpp113 std::vector<Tensor> dst_tensors; in _broadcast_out_impl() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_model.cc755 std::vector<DummySpatialTensor> dst_tensors(node.outputs.size()); in ResolvePolymorphicArgs() local