Searched refs:num_tensors_ (Results 1 – 3 of 3) sorted by relevance
40 num_tensors_ = usage_records.size(); in Build()41 source_ = 2 * num_tensors_; in Build()133 assignment->object_ids.assign(num_tensors_, kNotAssigned); in CalculateAssignment()134 is_tensor_assigned_.resize(num_tensors_); in CalculateAssignment()165 return vertex_id >= num_tensors_ && vertex_id < 2 * num_tensors_; in IsRightPartVertex()171 return vertex_id - num_tensors_; in LeftPartTwin()174 return vertex_id + num_tensors_; in RightPartTwin()203 size_t num_tensors_; member in tflite::gpu::__anondc65f6650111::MinCostFlowSolver
79 num_tensors_(input->output_dtypes().size()), in FileDatasetBase()80 tensor_index_padding_size_(StringPaddingSize(num_tensors_)), in FileDatasetBase()284 if (out_tensors->size() != dataset()->num_tensors_) { in GetNextInternal()288 dataset()->num_tensors_, " got: ", out_tensors->size()); in GetNextInternal()292 DCHECK_LT(tensor_index, dataset()->num_tensors_); in GetNextInternal()508 out_tensors->resize(dataset()->num_tensors_); in GetNextInternal()510 for (size_t i = 0; i < dataset()->num_tensors_; ++i) { in GetNextInternal()610 const size_t num_tensors_; member in tensorflow::data::CacheDatasetOp::FileDatasetBase
383 num_tensors_ = variants_.size(); in InitializeFromIterator()405 num_tensors_ = data.size(); in InitFromTensor()409 int64 NumTensors() { return num_tensors_; } in NumTensors()435 int64 num_tensors_; member in tensorflow::data::__anonaa5420280511::IteratorVariantSerializer