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