Searched defs:XlaGather (Results 1 – 2 of 2) sorted by relevance
35 Status XlaGather(const xla::XlaOp& input, const TensorShape& input_shape, in XlaGather() function
29391 func XlaGather(scope *Scope, operand tf.Output, start_indices tf.Output, slice_sizes tf.Output, dim… func