Searched defs:parent_shape (Results 1 – 3 of 3) sorted by relevance
42 TensorShape extend_parent_shape(TensorShape parent_shape, TensorShape shape, Coordinates coords) in extend_parent_shape()
155 … const TensorShape &parent_shape, const Coordinates &coords, const TensorShape &shape) in error_on_invalid_subtensor()
41 const TensorShape &parent_shape = parent_info.tensor_shape(); in validate_subtensor_shape() local