Searched refs:push_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 31 Status BuildTensorList(const xla::XlaOp& buffer, const xla::XlaOp& push_index, in BuildTensorList() argument
34 *output_list = xla::Tuple(buffer.builder(), {buffer, push_index}); in BuildTensorList()
44 Status GetTensorListPushIndex(const xla::XlaOp& op, xla::XlaOp* push_index) { in GetTensorListPushIndex() argument
46 *push_index = xla::GetTupleElement(op, 1); in GetTensorListPushIndex()
95 xla::XlaOp push_index; in InitializeTensorList() local
96 TF_RETURN_IF_ERROR(GetTensorListPushIndex(uninitialized_list, &push_index)); in InitializeTensorList()
97 return BuildTensorList(new_buffer, push_index, output_list); in InitializeTensorList()
|
| D | aggregate_ops.cc | 58 xla::XlaOp push_index; in Compile() local
59 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(ctx->Input(0), &push_index)); in Compile()
60 OP_REQUIRES_OK(ctx, BuildTensorList(sum, push_index, &sum)); in Compile()
|
| D | tensor_list_utils.h | 35 Status BuildTensorList(const xla::XlaOp& buffer, const xla::XlaOp& push_index,
42 Status GetTensorListPushIndex(const xla::XlaOp& op, xla::XlaOp* push_index);
|
| D | tensor_list_ops.cc | 355 xla::XlaOp push_index; in Compile() local
356 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(tl, &push_index)); in Compile()
373 push_index, &output_list)); in Compile()
|