Home
last modified time | relevance | path

Searched defs:weights_layout (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/examples/
H A Dgraph_inception_v4.cpp77 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
197 ConcatLayer get_mixed_3a(const std::string &data_path, DataLayout weights_layout) in get_mixed_3a()
222 ConcatLayer get_mixed_4a(const std::string &data_path, DataLayout weights_layout) in get_mixed_4a()
299 ConcatLayer get_mixed_5a(const std::string &data_path, DataLayout weights_layout) in get_mixed_5a()
324 …ConcatLayer get_inceptionA_block(const std::string &data_path, DataLayout weights_layout, std::str… in get_inceptionA_block()
419 ConcatLayer get_reductionA_block(const std::string &data_path, DataLayout weights_layout) in get_reductionA_block()
479 …ConcatLayer get_inceptionB_block(const std::string &data_path, DataLayout weights_layout, std::str… in get_inceptionB_block()
607 ConcatLayer get_reductionB_block(const std::string &data_path, DataLayout weights_layout) in get_reductionB_block()
689 …ConcatLayer get_inceptionC_block(const std::string &data_path, DataLayout weights_layout, std::str… in get_inceptionC_block()
H A Dgraph_inception_resnet_v2.cpp87 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
218 void block_mixed_5b(const std::string &data_path, DataLayout weights_layout) in block_mixed_5b()
321 void block_mixed_6a(const std::string &data_path, DataLayout weights_layout) in block_mixed_6a()
385 void block_mixed_7a(const std::string &data_path, DataLayout weights_layout) in block_mixed_7a()
488 …void block35_repeat(const std::string &data_path, DataLayout weights_layout, unsigned int num_bloc… in block35_repeat()
599 …void block17_repeat(const std::string &data_path, DataLayout weights_layout, unsigned int num_bloc… in block17_repeat()
683 …void block8_repeat(const std::string &data_path, DataLayout weights_layout, unsigned int num_block… in block8_repeat()
H A Dgraph_inception_resnet_v1.cpp103 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
239 …void block35_repeat(const std::string &data_path, DataLayout weights_layout, unsigned int num_bloc… in block35_repeat()
350 …void block17_repeat(const std::string &data_path, DataLayout weights_layout, unsigned int num_bloc… in block17_repeat()
434 …void block8_repeat(const std::string &data_path, DataLayout weights_layout, unsigned int num_block… in block8_repeat()
537 void reduction_a(const std::string &data_path, DataLayout weights_layout) in reduction_a()
601 void reduction_b(const std::string &data_path, DataLayout weights_layout) in reduction_b()
H A Dgraph_inception_v3.cpp75 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
226 …inception_node_A(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_inception_node_A()
356 …inception_node_B(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_inception_node_B()
427 …inception_node_C(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_inception_node_C()
588 …inception_node_D(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_inception_node_D()
687 …inception_node_E(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_inception_node_E()
H A Dgraph_yolov3.cpp76 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
422 … std::pair<SubStream, SubStream> darknet53(const std::string &data_path, DataLayout weights_layout) in darknet53()
537 … darknet53_block(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in darknet53_block()
H A Dgraph_squeezenet_v1_1.cpp74 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
191 …expand_fire_node(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_expand_fire_node()
H A Dgraph_squeezenet.cpp74 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
191 …expand_fire_node(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_expand_fire_node()
H A Dgraph_resnext50.cpp73 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
124 void add_residual_block(const std::string &data_path, DataLayout weights_layout, in add_residual_block()
H A Dgraph_resnet12.cpp91 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
160 …dd_residual_block(const std::string &data_path, const std::string &name, DataLayout weights_layout) in add_residual_block()
H A Dgraph_shufflenet.cpp89 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
170 void add_residual_block(const std::string &data_path, DataLayout weights_layout, in add_residual_block()
H A Dgraph_googlenet.cpp77 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
151 …t_inception_node(const std::string &data_path, std::string &&param_path, DataLayout weights_layout, in get_inception_node()
H A Dgraph_resnet50.cpp75 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
139 …dd_residual_block(const std::string &data_path, const std::string &name, DataLayout weights_layout, in add_residual_block()
H A Dgraph_resnet_v2_50.cpp78 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
142 …dd_residual_block(const std::string &data_path, const std::string &name, DataLayout weights_layout, in add_residual_block()
H A Dgraph_lenet.cpp74 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
H A Dgraph_vgg19.cpp73 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
H A Dgraph_alexnet.cpp81 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
H A Dgraph_vgg16.cpp74 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
H A Dgraph_srcnn955.cpp88 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
H A Dgraph_vgg_vdsr.cpp89 const DataLayout weights_layout = DataLayout::NCHW; in do_setup() local
H A Dgraph_deepspeech_v0_4_1.cpp84 const DataLayout weights_layout = DataLayout::NHWC; in do_setup() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_generic.h72 WeightsLayout weights_layout; member