Searched refs:raw_var_shape (Results 1 – 1 of 1) sorted by relevance
2689 xla::StatusOr<Node*> CreatePerHostDummyArgs(const InferredShape& raw_var_shape, in CreatePerHostDummyArgs() argument2704 if (!raw_var_shape.handle_shape.AsTensorShape(&var_shape) && in CreatePerHostDummyArgs()2705 !raw_var_shape.shape.AsTensorShape(&var_shape)) { in CreatePerHostDummyArgs()