Searched refs:lengths_tensor (Results 1 – 3 of 3) sorted by relevance
487 Tensor* lengths_tensor = nullptr; in Compute() local491 &lengths_tensor)); in Compute()492 auto lengths_tensor_vec = lengths_tensor->vec<int64_t>(); in Compute()
863 Tensor* lengths_tensor = nullptr; in Compute() local867 &lengths_tensor)); in Compute()868 auto lengths_tensor_t = lengths_tensor->vec<int64_t>(); in Compute()
310 tensorflow::Tensor lengths_tensor(tensorflow::DT_INT64, in HandleTensorArrayConcatV3Op() local313 lengths_tensor.vec<int64_t>()(i) = buffer_type.getDimSize(1); in HandleTensorArrayConcatV3Op()317 tensorflow::ConvertTensor(lengths_tensor, &builder).ValueOrDie())); in HandleTensorArrayConcatV3Op()