Searched refs:IsNestedTensorList (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 305 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
363 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
401 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
452 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
483 OP_REQUIRES_OK(ctx, IsNestedTensorList(input, &is_nested)); in Compile()
621 OP_REQUIRES_OK(ctx, IsNestedTensorList(initialized_list, &is_nested)); in Compile()
|
| D | aggregate_ops.cc | 51 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested_list)); in Compile()
|
| D | tensor_list_utils.cc | 121 Status IsNestedTensorList(xla::XlaOp list, bool* is_nested_list) { in IsNestedTensorList() function
426 TF_RETURN_IF_ERROR(IsNestedTensorList(list, element_is_tensor_list)); in ExecuteTensorListPopBack()
473 TF_RETURN_IF_ERROR(IsNestedTensorList(list, &is_nested)); in ExecuteTensorListSetItem()
509 TF_RETURN_IF_ERROR(IsNestedTensorList(list, &is_nested)); in ExecuteTensorListGetItem()
|
| D | tensor_list_utils.h | 34 Status IsNestedTensorList(xla::XlaOp list, bool* is_nested_list);
|