Lines Matching refs:FlatMap
50 using TensorTape = gtl::FlatMap<int64, int64>;
54 using OpTape = gtl::FlatMap<int64, OpTapeEntry<BackwardFunction, TapeTensor>>;
146 const gtl::FlatMap<int64, TapeTensor> sources_that_are_targets,
159 gtl::FlatMap<int64, int64> tensor_usage_;
310 gtl::FlatMap<int64, int64> tensor_usage_counts;
314 gtl::FlatMap<int64, int64> op_missing_tensor;
386 const gtl::FlatMap<int64, int64>& op_missing_tensor) { in InitialStack()
400 gtl::FlatMap<int64, TapeTensor> sources_that_are_targets, in InitialGradients()
403 gtl::FlatMap<int64, std::vector<Gradient*>>* result) { in InitialGradients()
457 gtl::FlatMap<string, gtl::FlatSet<int>>* FunctionsAcceptingNoneForIndicesMap() { in FunctionsAcceptingNoneForIndicesMap()
458 static auto* const m = new gtl::FlatMap<string, gtl::FlatSet<int>>({ in FunctionsAcceptingNoneForIndicesMap()
479 const gtl::FlatMap<int64, TapeTensor> sources_that_are_targets, in ComputeGradient()
488 gtl::FlatMap<int64, std::vector<Gradient*>> gradients; in ComputeGradient()
504 gtl::FlatMap<int64, int64> gradients_size; in ComputeGradient()