Home
last modified time | relevance | path

Searched refs:push_index (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc132 Status BuildNonNestedTensorList(xla::XlaOp buffer, xla::XlaOp push_index, in BuildNonNestedTensorList() argument
135 *output_list = xla::Tuple(buffer.builder(), {buffer, push_index}); in BuildNonNestedTensorList()
160 Status GetTensorListPushIndex(xla::XlaOp list, xla::XlaOp* push_index) { in GetTensorListPushIndex() argument
168 *push_index = xla::GetTupleElement(list, tuple_size - 1); in GetTensorListPushIndex()
172 Status SetTensorListPushIndex(xla::XlaOp list, xla::XlaOp push_index, in SetTensorListPushIndex() argument
186 result_parts.push_back(push_index); in SetTensorListPushIndex()
364 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPushBack() local
383 start_indices[0] = push_index; in ExecuteTensorListPushBack()
399 start_indices[0] = push_index; in ExecuteTensorListPushBack()
407 xla::XlaOp updated_push_index = push_index + xla::ConstantR0<int32>(b, 1); in ExecuteTensorListPushBack()
[all …]
Dtensor_list_utils.h37 Status BuildNonNestedTensorList(xla::XlaOp buffer, xla::XlaOp push_index,
53 Status GetTensorListPushIndex(xla::XlaOp list, xla::XlaOp* push_index);
58 Status SetTensorListPushIndex(xla::XlaOp list, xla::XlaOp push_index,
122 Status ExecuteTensorListFromTensor(int push_index, xla::XlaOp tensor,
Daggregate_ops.cc75 xla::XlaOp push_index; in Compile() local
76 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(ctx->Input(0), &push_index)); in Compile()
77 OP_REQUIRES_OK(ctx, BuildNonNestedTensorList(sum, push_index, &sum)); in Compile()
Dshape_op.cc372 xla::XlaOp push_index; in Compile() local
373 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(list, &push_index)); in Compile()
377 SetTensorListPushIndex(new_list, push_index, &result)); in Compile()