Searched refs:lengths_tensor (Results 1 – 3 of 3) sorted by relevance
381 Tensor* lengths_tensor = nullptr; in Compute() local386 &lengths_tensor)); in Compute()387 auto lengths_tensor_vec = lengths_tensor->vec<int64>(); in Compute()
869 Tensor* lengths_tensor = nullptr; in Compute() local872 &lengths_tensor)); in Compute()873 auto lengths_tensor_t = lengths_tensor->vec<int64>(); in Compute()
311 tensorflow::Tensor lengths_tensor(tensorflow::DT_INT64, in HandleTensorArrayConcatV3Op() local314 lengths_tensor.vec<tensorflow::int64>()(i) = buffer_type.getDimSize(1); in HandleTensorArrayConcatV3Op()318 tensorflow::ConvertTensor(lengths_tensor, &builder).ValueOrDie())); in HandleTensorArrayConcatV3Op()