Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Daggregate_ops.cc77 OP_REQUIRES_OK(ctx, BuildNonNestedTensorList(sum, push_index, &sum)); in Compile()
H A Dtensor_list_utils.h37 Status BuildNonNestedTensorList(xla::XlaOp buffer, xla::XlaOp push_index,
H A Dtensor_list_utils.cc132 Status BuildNonNestedTensorList(xla::XlaOp buffer, xla::XlaOp push_index, in BuildNonNestedTensorList() function