Searched defs:gather_index (Results 1 – 2 of 2) sorted by relevance
1913 def gather_index(self): member in _LayerBroadcaster1936 def from_gather_index(cls, gather_index): argument2063 def __init__(self, gather_index): argument2071 def gather_index(self): member in _GatherLayerBroadcaster
401 int64_t gather_index = dense_spec.final_shape_gather_indices[dense_dim]; in ValidateStridedSliceOp() local