Searched refs:finfer_shape (Results 1 – 14 of 14) sorted by relevance
44 static auto& finfer_shape = in InferAttr() local167 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()
168 static auto& finfer_shape = Op::GetAttr<FInferShape>("FInferShape"); in Gradient() local221 if (finfer_shape.contains(git->node->op())) { in Gradient()
38 static auto& finfer_shape = Op::GetAttr<FInferNodeEntryAttr<AttrType>>(infer_name); in InferAttr() local157 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()
163 static auto& finfer_shape = Op::GetAttr<FInferShape>("FInferShape"); in Gradient() local215 if (finfer_shape.contains(git->node->op())) { in Gradient()
169 static auto& finfer_shape = Op::GetAttr<mxnet::FInferShape>("FInferShape"); in Gradient() local222 if (finfer_shape.contains(git->node->op())) { in Gradient()
289 static auto& finfer_shape = in InferAttr() local421 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()574 static auto& finfer_shape = in InferShapeAttr() local724 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferShapeAttr()
1431 static auto& finfer_shape = nnvm::Op::GetAttr<mxnet::FInferShape>("FInferShape"); in RunOps() local1490 if (finfer_shape.count(op)) { in RunOps()1499 auto finfer = finfer_shape[op]; in RunOps()
470 static auto& finfer_shape = Op::GetAttr<mxnet::FInferShape>("FInferShape"); variable471 if (finfer_shape.count(op_)) {472 mxnet::FInferShape call_infer_shapes = finfer_shape[op_];