Searched defs:NodeContext (Results 1 – 2 of 2) sorted by relevance
74 struct NodeContext { struct75 Node* node;76 std::vector<Value*> inputs;77 std::vector<Value*> outputs;
659 struct NodeContext { struct in tensorflow::grappler::SymbolicShapeRefiner660 const OpRegistrationData* op_data;661 DataTypeVector input_types;662 DataTypeVector output_types;663 std::unique_ptr<InferenceContext> inference_context;665 std::vector<const TensorProto*> input_tensor_protos;666 std::vector<const TensorProto*> output_tensor_protos;669 std::vector<ShapeHandle> input_tensors_as_shapes_to_propagate;670 std::vector<ShapeHandle> output_tensors_as_shapes;673 bool shape_incompatible = false;[all …]