Home
last modified time | relevance | path

Searched defs:push_index (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc31 Status BuildTensorList(const xla::XlaOp& buffer, const xla::XlaOp& push_index, in BuildTensorList()
44 Status GetTensorListPushIndex(const xla::XlaOp& op, xla::XlaOp* push_index) { in GetTensorListPushIndex()
95 xla::XlaOp push_index; in InitializeTensorList() local
Daggregate_ops.cc58 xla::XlaOp push_index; in Compile() local
Dtensor_list_ops.cc355 xla::XlaOp push_index; in Compile() local