Searched defs:transpose_param (Results 1 – 8 of 8) sorted by relevance
55 auto transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineBNPattern() local75 auto transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineActivationscalePattern() local93 auto transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineActivationPattern() local107 auto transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineBiasAddPattern() local145 auto transpose_param = std::make_shared<CondVar>(IsParamOrValueNodeWithData); in DefineTransTransPattern() local
31 auto transpose_param = op_->mutable_transpose(); in BuildLayer() local
29 internal::TransposeParam transpose_param; in CreateOpParam() local
106 TransposeParameter *transpose_param = (TransposeParameter *)parameter; in TransposeInferShape() local
176 const TransposeParameter *transpose_param) { in DoTransposeInt8()225 const TransposeParameter *transpose_param, int task_id, int thread_num) { in TransposeDimsInt8()
245 const TransposeParameter *transpose_param) { in DoTranspose()508 … size_t count, const TransposeParameter *transpose_param) { in ParallelRun()526 … const TransposeParameter *transpose_param, int task_id, in TransposeDims()
141 const TransposeParameter *transpose_param) { in DoTranspose()398 … const TransposeParameter *transpose_param, int task_id, int thread_num) { in TransposeDims()
174 …NNaclFp32Serializer::CodeStruct(const std::string &name, const TransposeParameter &transpose_param, in CodeStruct()