Searched defs:XlaArgument (Results 1 – 1 of 1) sorted by relevance
31 struct XlaArgument { struct32 enum Kind {56 Kind kind = kInvalid;60 DataType type = DT_INVALID;73 absl::variant<TensorShape, xla::Shape> shape;77 Tensor constant_value;80 absl::optional<Tensor> value_bound;83 string name;86 string node_name;89 XlaResource::Kind resource_kind = XlaResource::kInvalid;[all …]