Searched refs:IsNestedTensorList (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 299 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
355 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
393 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
444 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested)); in Compile()
475 OP_REQUIRES_OK(ctx, IsNestedTensorList(input, &is_nested)); in Compile()
613 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
424 TF_RETURN_IF_ERROR(IsNestedTensorList(list, element_is_tensor_list)); in ExecuteTensorListPopBack()
471 TF_RETURN_IF_ERROR(IsNestedTensorList(list, &is_nested)); in ExecuteTensorListSetItem()
507 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);
|