Searched defs:NodeContext (Results 1 – 1 of 1) sorted by relevance
653 struct NodeContext { struct in tensorflow::grappler::SymbolicShapeRefiner654 const OpRegistrationData* op_data;655 DataTypeVector input_types;656 DataTypeVector output_types;657 std::unique_ptr<InferenceContext> inference_context;659 std::vector<const TensorProto*> input_tensor_protos;660 std::vector<const TensorProto*> output_tensor_protos;663 std::vector<ShapeHandle> input_tensors_as_shapes_to_propagate;664 std::vector<ShapeHandle> output_tensors_as_shapes;667 bool shape_incompatible = false;[all …]