Searched refs:input_shape_pre_axis (Results 1 – 1 of 1) sorted by relevance
69 TensorShape input_shape_pre_axis{input_shape}; in XlaGather() local70 input_shape_pre_axis.RemoveDimRange(axis, input_shape.dims()); in XlaGather()80 out_shape.AppendShape(input_shape_pre_axis); in XlaGather()