Home
last modified time | relevance | path

Searched refs:delimiter_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/
H A Dstring_split_fuzz.cc34 Tensor delimiter_tensor(tensorflow::DT_STRING, TensorShape({})); in FuzzImpl() local
45 delimiter_tensor.scalar<tstring>()() = in FuzzImpl()
50 RunInputs({{"input", input_tensor}, {"delimiter", delimiter_tensor}}); in FuzzImpl()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstring_split_op.cc184 const Tensor* delimiter_tensor; in Compute() local
185 OP_REQUIRES_OK(ctx, ctx->input("delimiter", &delimiter_tensor)); in Compute()
187 ctx, TensorShapeUtils::IsScalar(delimiter_tensor->shape()), in Compute()
189 delimiter_tensor->shape().DebugString())); in Compute()
190 const auto delimiter_vec = delimiter_tensor->flat<tstring>(); in Compute()