Searched defs:input_tensor_desc (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/graph/ |
| H A D | GraphBuilder.cpp | 158 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_batch_normalization_node() local
233 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_convolution_node() local
288 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_deconvolution_node() local
342 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_depthwise_convolution_node() local
479 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_fully_connected_layer() local
506 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_fully_connected_layer() local
571 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_normalize_planar_yuv_node() local
685 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_scale_layer() local
740 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in add_yolo_node() local
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | cudnn_extension.cpp | 50 torch::native::TensorDescriptor input_tensor_desc(inputs, 4); in cudnn_relu() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | GroupedConvolutionMutator.cpp | 51 …const TensorDescriptor input_tensor_desc = get_tensor_descriptor(g, g.node(input.node_id)->outputs… in create_grouped_convolution() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 5940 CudnnTensorDescriptor input_tensor_desc( in DoTransformTensor() local
|