Searched refs:indices_are_vectors (Results 1 – 3 of 3) sorted by relevance
35 const xla::XlaOp& indices, bool indices_are_vectors, in XlaScatter() argument48 if (indices_are_vectors) { in XlaScatter()139 dim_numbers.set_index_vector_dim(indices_are_vectors in XlaScatter()
48 const xla::XlaOp& indices, bool indices_are_vectors,
150 OpKernelConstruction* context, bool indices_are_vectors, in ResourceScatterOp() argument155 indices_are_vectors_(indices_are_vectors), in ResourceScatterOp()