Searched refs:size_func_ (Results 1 – 3 of 3) sorted by relevance
52 : size_func_(size_func) {} in OptimizeInputOutputBufferAlias()66 ShapeSizeFunction size_func_ = nullptr; variable
56 int64 bytes = size_func_(subshape); in Build()67 int64 bytes = size_func_(subshape); in Build()
116 std::function<int64(const Shape&)> size_func_) const;