Home
last modified time | relevance | path

Searched defs:NodeContext (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc562 struct NodeContext { struct in tensorflow::grappler::SymbolicShapeRefiner
563 const OpRegistrationData* op_data;
564 DataTypeVector input_types;
565 DataTypeVector output_types;
566 std::unique_ptr<InferenceContext> inference_context;
568 std::vector<const TensorProto*> input_tensor_protos;
569 std::vector<const TensorProto*> output_tensor_protos;
570 std::vector<ShapeHandle> output_tensors_as_shapes;