Searched defs:tensor_node (Results 1 – 9 of 9) sorted by relevance
92 auto tensor_node = CreateScalarValueTensor(func_graph, make_tuple_input); in Process() local95 auto tensor_node = CreateScalarToTensor(func_graph, make_tuple_input, type_id); in Process() local
89 AnfNodePtr tensor_node = input; in Process() local
62 auto tensor_node = std::make_shared<ValueNode>(tensor_ptr); in Process() local
143 … auto tensor_node = std::dynamic_pointer_cast<tensor::Tensor>(node->cast<ValueNodePtr>()->value()); in Ref() local162 auto tensor_node = std::dynamic_pointer_cast<tensor::Tensor>(value); in Ref() local
281 CNodePtr NewCastNode(const AnfNodePtr &tensor_node, const TypeId &dtype) { in NewCastNode()294 CNodePtr NewTransposeNode(const AnfNodePtr &tensor_node, const AnfNodePtr &tuple, ShapeVector outpu… in NewTransposeNode()305 CNodePtr NewTileNode(const AnfNodePtr &tensor_node, const AnfNodePtr &tuple) { in NewTileNode()
235 for (const auto &tensor_node : tensor_to_node) { in UpdateDynamicOutputShape() local
397 for (auto &tensor_node : task->tensor_to_node_) { in RunGraphAsync() local
2820 auto tensor_node = NewValueNode(tensor_ptr); in CreateTensorTupleAnfNodePtr() local