Searched refs:input_shape_post_axis (Results 1 – 1 of 1) sorted by relevance
71 TensorShape input_shape_post_axis{input_shape}; in XlaGather() local72 input_shape_post_axis.RemoveDimRange(0, axis + num_index_dims); in XlaGather()82 out_shape.AppendShape(input_shape_post_axis); in XlaGather()