Searched defs:tensor_to_scalar (Results 1 – 5 of 5) sorted by relevance
40 VectorRef tensor_to_scalar({std::make_shared<Primitive>(kTensorToScalarOpName), x, dtype}); in DefinePattern() local
78 CNodePtr tensor_to_scalar = CreateTensorToScalar(graph, strided_slice_node); in Process() local104 CNodePtr tensor_to_scalar = func_graph->NewCNode(inputs); in CreateTensorToScalar() local
163 CNodePtr tensor_to_scalar = func_graph->NewCNode(inputs); in GenerateTensorToScalar() local315 CNodePtr tensor_to_scalar = GenerateTensorToScalar(func_graph, replace_op, true); in ReplaceScalarOp() local436 CNodePtr tensor_to_scalar = GenerateTensorToScalar(func_graph, strided_slice_node, true); in ReplaceShapeTupleGet() local
548 …auto tensor_to_scalar = func_graph->get_return()->input(1)->cast<CNodePtr>()->input(1)->cast<CNode… in TEST_F() local
940 CNodePtr tensor_to_scalar = func_graph->NewCNode(inputs); in ProcessTensorToScalar() local