Home
last modified time | relevance | path

Searched refs:indices_are_vectors (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc35 const xla::XlaOp& indices, bool indices_are_vectors, in XlaScatter() argument
48 if (indices_are_vectors) { in XlaScatter()
139 dim_numbers.set_index_vector_dim(indices_are_vectors in XlaScatter()
Dscatter.h48 const xla::XlaOp& indices, bool indices_are_vectors,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dvariable_ops.cc150 OpKernelConstruction* context, bool indices_are_vectors, in ResourceScatterOp() argument
155 indices_are_vectors_(indices_are_vectors), in ResourceScatterOp()