Home
last modified time | relevance | path

Searched refs:IsTensorListInitialized (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc115 Status IsTensorListInitialized(xla::XlaOp list, bool* is_initialized) { in IsTensorListInitialized() function
123 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in IsNestedTensorList()
141 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in GetTensorListBufferShape()
152 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in GetTensorListBuffer()
162 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in GetTensorListPushIndex()
175 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in SetTensorListPushIndex()
209 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in GetLeadingDimForTensorList()
317 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in GetInitializedTensorListForElement()
356 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in ExecuteTensorListPushBack()
418 TF_RETURN_IF_ERROR(IsTensorListInitialized(list, &is_initialized)); in ExecuteTensorListPopBack()
[all …]
Dtensor_list_ops.cc293 (IsTensorListInitialized(ctx->Input(0), &is_initialized))); in Compile()
349 (IsTensorListInitialized(ctx->Input(0), &is_initialized))); in Compile()
387 (IsTensorListInitialized(ctx->Input(0), &is_initialized))); in Compile()
438 (IsTensorListInitialized(ctx->Input(0), &is_initialized))); in Compile()
469 OP_REQUIRES_OK(ctx, (IsTensorListInitialized(input, &is_initialized))); in Compile()
667 (IsTensorListInitialized(ctx->Input(0), &is_initialized))); in Compile()
Daggregate_ops.cc43 OP_REQUIRES_OK(ctx, IsTensorListInitialized(list, &is_initialized)); in Compile()
Dtensor_list_utils.h29 Status IsTensorListInitialized(xla::XlaOp list, bool* is_initialized);
Dshape_op.cc346 OP_REQUIRES_OK(ctx, IsTensorListInitialized(list, &is_initialized)); in Compile()
Dwhile_op.cc106 IsTensorListInitialized(ctx->Input(i), &arg.initialized)); in MakeXlaCompilerArgumentsFromInputs()