Searched defs:keep_dim (Results 1 – 8 of 8) sorted by relevance
22 bool keep_dim; member in add_reduce_kernel_attr48 bool keep_dim = extra->Attr<bool>("keep_dim"); in CustomKernelInferShape() local80 bool keep_dim = kernel_ptr->keep_dim; in CustomKernel() local
36 auto keep_dim = PytorchNodeParser::GetValueFromConstNode<bool>(torch_node->input(THIRD_INPUT)); in Parse() local
28 def __init__(self, func_path, out_types, axis, keep_dim): argument
122 auto keep_dim = py::bool_(true); in TEST_F() local150 auto keep_dim = py::bool_(false); in TEST_F() local
105 bool keep_dim = in CheckReduce() local
346 bool keep_dim, bool skip_mode) const { in NeedReduce()388 …dePtr> Emitter::NeedReduce(const NodePtr &shape, const NodePtr &axis, bool keep_dim, bool skip_mod… in NeedReduce()
42 def _get_reshape_shape(shape, keep_dim=0): argument
305 …bool keep_dim = prim->GetAttr(ops::kKeepDims) != nullptr && GetValue<bool>(prim->GetAttr(ops::kKee… in ReduceInferShape() local