Searched defs:NodeContext (Results 1 – 1 of 1) sorted by relevance
630 struct NodeContext { struct in tensorflow::grappler::SymbolicShapeRefiner631 const OpRegistrationData* op_data;632 DataTypeVector input_types;633 DataTypeVector output_types;634 std::unique_ptr<InferenceContext> inference_context;636 std::vector<const TensorProto*> input_tensor_protos;637 std::vector<const TensorProto*> output_tensor_protos;640 std::vector<ShapeHandle> input_tensors_as_shapes_to_propagate;641 std::vector<ShapeHandle> output_tensors_as_shapes;644 bool shape_incompatible = false;[all …]