Searched refs:element_tensor (Results 1 – 2 of 2) sorted by relevance
169 tensorflow::Tensor element_tensor(ctx->allocator({}), in GetNextInternal() local171 element_tensor.scalar<tensorflow::tstring>()() = element; in GetNextInternal()174 out_tensors->push_back(std::move(element_tensor)); in GetNextInternal()
537 const Tensor& element_tensor = tensor_list->tensors()[i]; in Compute() local538 if (element_tensor.dtype() != DT_INVALID) { in Compute()539 if (element_tensor.NumElements() > 0) { in Compute()541 element_tensor.shaped<T, 2>({1, element_tensor.NumElements()}))); in Compute()