Searched refs:GetTensorListPushIndex (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 44 Status GetTensorListPushIndex(const xla::XlaOp& op, xla::XlaOp* push_index) { in GetTensorListPushIndex() function
96 TF_RETURN_IF_ERROR(GetTensorListPushIndex(uninitialized_list, &push_index)); in InitializeTensorList()
|
| D | tensor_list_ops.cc | 51 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(ctx->Input(0), &index)); in Compile()
356 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(tl, &push_index)); in Compile()
401 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(list_tuple, &index)); in Compile()
445 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(state, &index)); in Compile()
|
| D | tensor_list_utils.h | 42 Status GetTensorListPushIndex(const xla::XlaOp& op, xla::XlaOp* push_index);
|
| D | aggregate_ops.cc | 59 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(ctx->Input(0), &push_index)); in Compile()
|