Searched refs:node_shape (Results 1 – 2 of 2) sorted by relevance
82 const PartialTensorShape node_shape(node_value.tensor().tensor_shape()); in ValidateNode() local83 if (node_shape.num_elements() < 0) { in ValidateNode()87 node_shape.num_elements(), " elements"); in ValidateNode()
710 string node_shape = "circle"; in DumpRootTag() local723 to_id, node_body, node_shape, NodeColorAttributes(color)); in DumpRootTag()790 string node_shape = GetInstructionNodeShape(instr); in DumpInstruction() local805 node_shape = "diamond"; in DumpInstruction()820 InstructionId(instr), node_body, node_shape, node_metadata, in DumpInstruction()